Метод concatenate

Метод concatenate класса Dataset создает новый датасет, который содержит все элементы текущего датасета, за которыми следуют все элементы переданного датасета. Метод принимает один обязательный параметр - другой объект Dataset, который нужно присоединить. Оба датасета должны иметь совместимые структуры элементов.

Синтаксис

dataset.concatenate(other_dataset)

Пример

Давайте создадим два датасета из тензоров и объединим их:

import tensorflow as tf

ds1 = tf.data.Dataset.from_tensor_slices([1, 2, 3])
ds2 = tf.data.Dataset.from_tensor_slices([4, 5, 6])

ds = ds1.concatenate(ds2)

for elem in ds:
	print(elem.numpy())

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

Пример

Давайте объединим датасеты, содержащие двумерные тензоры:

import tensorflow as tf

ds1 = tf.data.Dataset.from_tensor_slices([[1, 2, 3], [4, 5, 6]])
ds2 = tf.data.Dataset.from_tensor_slices([[7, 8, 9]])

ds = ds1.concatenate(ds2)

for elem in ds:
	print(elem.numpy())

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

[1 2 3]
[4 5 6]
[7 8 9]

Пример

Давайте объединим датасеты строковых данных:

import tensorflow as tf

ds1 = tf.data.Dataset.from_tensor_slices(['abcde', 'model.keras'])
ds2 = tf.data.Dataset.from_tensor_slices(['xyz', 'test'])

ds = ds1.concatenate(ds2)

for elem in ds:
	print(elem.numpy().decode('utf-8'))

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

abcde
model.keras
xyz
test

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

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