Метод take класса Dataset

Метод take класса Dataset создает новый датасет, который содержит только первые count элементов исходного датасета. Если в исходном датасете меньше элементов, чем указано в параметре count, то возвращается датасет со всеми имеющимися элементами. Этот метод часто используется для быстрого просмотра или отладки данных.

Синтаксис

dataset.take(count)

Пример

Давайте создадим датасет из тензора и возьмем первые три элемента:

import tensorflow as tf

dataset = tf.data.Dataset.from_tensor_slices([1, 2, 3, 4, 5])
res = dataset.take(3)

for element in res:
	print(element.numpy())

Результат выполнения кода:

1
2
3

Пример

Давайте попробуем взять больше элементов, чем есть в исходном датасете:

import tensorflow as tf

dataset = tf.data.Dataset.from_tensor_slices([1, 2, 3, 4, 5])
res = dataset.take(10)

for element in res:
	print(element.numpy())

Результат выполнения кода:

Смотрите также

метод skip,
который пропускает первые элементы датасета
метод batch,
который объединяет элементы датасета в пакеты
метод from_tensor_slices,
который создает датасет из тензора
метод take_while,
который берет элементы, пока выполняется условие