當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

Python tf.data.experimental.RandomDataset.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 張量在 0 和 len(datasets) - 1 之間。
  • stop_on_empty_dataset 如果 True ，如果遇到空數據集，選擇將停止。如果 False ，它會跳過空數據集。建議將其設置為 True 。否則，選定的元素會按照用戶的意圖開始，但可能會隨著輸入數據集變空而改變。這可能很難檢測到，因為數據集開始看起來是正確的。默認為 True 。

返回

  • 根據 choice_dataset 的值交織來自 datasets 的元素的數據集。

拋出

  • TypeError 如果datasets 或choice_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.experimental.RandomDataset.choose_from_datasets。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。