tf.contrib.data.choose_from_datasets(
datasets,
choice_dataset
)
Defined in tensorflow/contrib/data/python/ops/interleave_ops.py
.
Creates a dataset that deterministically chooses elements from datasets
. (deprecated)
For example, given the following datasets:
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.contrib.data.choose_from_datasets(datasets, choice_dataset)
The elements of result
will be:
"foo", "bar", "baz", "foo", "bar", "baz", "foo", "bar", "baz"
Args:
datasets
: A list oftf.data.Dataset
objects with compatible structure.choice_dataset
: Atf.data.Dataset
of scalartf.int64
tensors between0
andlen(datasets) - 1
.
Returns:
A dataset that interleaves elements from datasets
according to the values
of choice_dataset
.
Raises:
TypeError
: If thedatasets
orchoice_dataset
arguments have the wrong type.