
    BVh                     j    d 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.skip`.    )dataset_ops)dtypes)ops)gen_dataset_opsNc                     t        | ||      S N)_SkipDataset)selfcountnames      R/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/data/ops/skip_op.py_skipr      s    	dE4	((    c                   $     e Zd ZdZd fd	Z xZS )r	   z?A `Dataset` skipping the first `count` elements from its input.c                    || _         t        j                  |t        j                  d      | _        || _        t        j                  |j                  fd| j
                  i| j                  }t        | 1  ||       y)z!See `Dataset.skip()` for details.r   )dtyper   N)_input_datasetr   convert_to_tensorr   int64_count_namer   skip_dataset_variant_tensor_common_argssuper__init__)r
   input_datasetr   r   variant_tensor	__class__s        r   r   z_SkipDataset.__init__   sr    'D''V\\PDKDJ$11%%kk 

N 
G]N3r   r   )__name__
__module____qualname____doc__r   __classcell__)r   s   @r   r	   r	      s    G	4 	4r   r	   r   )r#   tensorflow.python.data.opsr   tensorflow.python.frameworkr   r   tensorflow.python.opsr   r   UnaryUnchangedStructureDatasetr	    r   r   <module>r*      s-    4 2 . + 1)4;== 4r   