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 | ||dS )N)name)_TakeDataset)selfcountr    r
   Q/var/www/myenv/lib/python3.10/site-packages/tensorflow/python/data/ops/take_op.py_take   s   r   c                       s"   e Zd ZdZd fdd	Z  ZS )r   zAA `Dataset` containing 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.take()` for details.r	   )dtyper   N)_input_datasetr   convert_to_tensorr   int64_count_namer   take_dataset_variant_tensor_common_argssuper__init__)r   input_datasetr	   r   variant_tensor	__class__r
   r   r      s   z_TakeDataset.__init__N)__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   
