
    BVh	                         d Z ddlZddlmZ ddlmZ ddlmZ ddlm	Z	 ddlm
Z
 ddlmZ dd	lmZ 	 	 	 dd
Z G d dej                         Zy)z/The implementation of `tf.data.Dataset.random`.    N)tf2)dataset_ops)random_seed)dtypes)tensor_spec)gen_dataset_ops)gen_experimental_dataset_opsc                     t        | ||      S )z#See `Dataset.random()` for details.seedrerandomize_each_iterationname)_RandomDatasetr   s      T/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/data/ops/random_op.py_randomr      s    
 
!;
     c                   4     e Zd ZdZd fd	Zed        Z xZS )r   #A `Dataset` of pseudorandom values.c                    t        j                  |      \  | _        | _        || _        || _        |r~t        j                         st        j                  d       t        j                  d| j                  | j                  t        j                         | j                  d| j                  }n6t        j                  d| j                  | j                  d| j                  }t         | E  |       y)r   z`In TF 1, the `rerandomize_each_iteration=True` option is only supported for repeat-based epochs.)r   seed2seed_generatorr   )r   r   N )r   get_seed_seed_seed2_rerandomize_namer   enabledwarningswarnged_opsrandom_dataset_v2r   dummy_seed_generator_common_argsrandom_datasetsuper__init__)selfr   r   r   variant_tensor	__class__s        r   r'   z_RandomDataset.__init__*   s    )2248DJ2DDJ![[] C 	D00 zz(==?%)%6%6	
 n -- CzzC040A0ACn	G^$r   c                 J    t        j                  g t        j                        S )N)r   
TensorSpecr   int64)r(   s    r   element_specz_RandomDataset.element_spec>   s    !!"fll33r   NNN)__name__
__module____qualname____doc__r'   propertyr.   __classcell__)r*   s   @r   r   r   '   s    +%( 4 4r   r   r/   )r3   r   tensorflow.pythonr   tensorflow.python.data.opsr   tensorflow.python.data.utilr   tensorflow.python.frameworkr   r   tensorflow.python.opsr   r	   r!   r   DatasetSourcer   r   r   r   <module>r<      sD    6  ! 2 3 . 3 1 I 
#	4[.. 4r   