当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python tf.data.TFRecordDataset.choose_from_datasets用法及代码示例

用法

@staticmethod
choose_from_datasets(
    datasets, choice_dataset, stop_on_empty_dataset=True
)

参数

  • datasets 具有兼容结构的tf.data.Dataset 对象的非空列表。
  • choice_dataset tf.data.Dataset 的标量 tf.int64 张量在 0len(datasets) - 1 之间。
  • stop_on_empty_dataset 如果 True ,如果遇到空数据集,选择将停止。如果 False ,它会跳过空数据集。建议将其设置为 True 。否则,选定的元素会按照用户的意图开始,但可能会随着输入数据集变空而改变。这可能很难检测到,因为数据集开始看起来是正确的。默认为 True

返回

  • 根据 choice_dataset 的值交织来自 datasets 的元素的数据集。

抛出

  • TypeError 如果datasetschoice_dataset 的类型错误。
  • ValueError 如果datasets 为空。

创建一个从 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.data.Dataset.choose_from_datasets(datasets, choice_dataset)

result 的元素将是:

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

相关用法

注:本文由堆栈答案筛选整理自tensorflow.org大神的英文原创作品 tf.data.TFRecordDataset.choose_from_datasets。非经特殊声明,原始代码版权归原作者所有,本译文的传播和使用请遵循“署名-相同方式共享 4.0 国际 (CC BY-SA 4.0)”协议。