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


Python tf.compat.v1.data.TFRecordDataset.repeat用法及代码示例


用法

repeat(
    count=None, name=None
)

参数

  • count (可选。)tf.int64 标量 tf.Tensor ,表示数据集应重复的次数。默认行为(如果 countNone-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.compat.v1.data.TFRecordDataset.repeat。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。