tf.data.experimental.choose_from_datasets

Создаёт набор данных, который детерминированно выбирает элементы из datasets.

tf.data.experimental.choose_from_datasets(
    datasets, choice_dataset
)

Например, при следующих наборах данных:

datasets = [tf.data.Dataset.from_tensors("foo").repeat(),
            tf.data.Dataset.from_tensors("bar").repeat(),
            tf.data.Dataset.from_tensors("baz").repeat()]

# Define a dataset containing `[0, 1, 2, 0, 1, 2, 0, 1, 2]`.
choice_dataset = tf.data.Dataset.range(3).repeat(3)

result = tf.data.experimental.choose_from_datasets(datasets, choice_dataset)

Элементы result будут:

"foo", "bar", "baz", "foo", "bar", "baz", "foo", "bar", "baz"

Аргументы
`datasets`	Список объектов `tf.data.Dataset` с совместимой структурой.
`choice_dataset`	`tf.data.Dataset` скалярных тензоров типа `tf.int64` в диапазоне от `0` до `len(datasets) - 1`.

Возвращает
Набор данных, который чередует элементы из `datasets` в соответствии со значениями `choice_dataset`.

Исключения
`TypeError`	Если аргументы `datasets` или `choice_dataset` имеют неправильный тип.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/data/experimental/choose_from_datasets