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

Python tf.data.TFRecordDataset.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.TFRecordDataset.repeat。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。