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