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

Python tf.data.experimental.RandomDataset.repeat用法及代碼示例


用法

repeat(
    count=None, name=None
)

參數

  • count (可選。)tf.int64 標量 tf.Tensor ，表示數據集應重複的次數。默認行為(如果 count 是 None 或 -1 )是無限期重複數據集。
  • name (可選。) tf.data 操作的名稱。

返回

  • Dataset 一個Dataset。

重複此數據集，以便看到每個原始值 count 次。

dataset = tf.data.Dataset.from_tensor_slices([1, 2, 3])
dataset = dataset.repeat(3)
list(dataset.as_numpy_iterator())
[1, 2, 3, 1, 2, 3, 1, 2, 3]

注意：如果這個數據集是全局狀態的函數(例如隨機數生成器)，那麽不同的重複可能會產生不同的元素。

相關用法


注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.data.experimental.RandomDataset.repeat。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。