o
    ?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   c/home/www/facesmatcher.com/pyenv/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	   )Zdtyper
   N)Z_input_datasetr   Zconvert_to_tensorr   Zint64_count_namer   Zskip_datasetZ_variant_tensorZ_common_argssuper__init__)r   Zinput_datasetr	   r
   Z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   Ztensorflow.python.data.opsr   Ztensorflow.python.frameworkr   r   Ztensorflow.python.opsr   r   ZUnaryUnchangedStructureDatasetr   r   r   r   r   <module>   s   
