Take

class tfdatacompose.take.Take(count: int)

Dataset take operation.

Wraps the Tensorflow Take operation on the dataset. Creates a new dataset by taking count element from the dataset.

Parameters:

count – the number of elements to take from the original dataset

apply(dataset: DatasetV1) → DatasetV1

Applies the operation on the given dataset and returns the new dataset.

Parameters:

dataset – Tensorflow Dataset The dataset to change

Returns:

Tensorflow Dataset The changed dataset