o
    i e                     @   sT   d Z ddlmZ ddlmZ ddlmZ ddlmZ dddZG d	d
 d
ej	Z
dS )z-The implementation of `tf.data.Dataset.skip`.    )dataset_ops)dtypes)ops)gen_dataset_opsNc                 C   s   t | ||S N)_SkipDataset)selfcountname r   Q/var/www/myenv/lib/python3.10/site-packages/tensorflow/python/data/ops/skip_op.py_skip   s   r   c                       s"   e Zd ZdZd fdd	Z  ZS )r   z?A `Dataset` skipping the first `count` elements from its input.Nc                    sN   || _ tj|tjdd| _|| _tj|j	fd| ji| j
}t || dS )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__r   r   r      s   z_SkipDataset.__init__r   )__name__
__module____qualname____doc__r   __classcell__r   r   r   r   r      s    r   r   )r    tensorflow.python.data.opsr   tensorflow.python.frameworkr   r   tensorflow.python.opsr   r   UnaryUnchangedStructureDatasetr   r   r   r   r   <module>   s   
