Tensorflow.js tf.data.Dataset.repeat()用法及代码示例

Tensorflow.js是Google开发的开源库，用于在浏览器或节点环境中运行机器学习模型和深度学习神经网络。它还可以帮助开发人员使用JavaScript语言开发ML模型，并可以直接在浏览器或Node.js中使用ML。

tf.data.Dataset.repeat() 函数用于重复数据集计数时间

用法：

repeat (count?)

参数

• count:一个整数，表示数据集应重复的次数。

返回值: 它返回 tf.data.Dataset。

范例1：

const a = tf.data.array([1, 2, 3]).repeat(3);
  
await a.forEachAsync(e => console.log(e)),

输出：

​1
2
3
1
2
3
1
2
3

范例2：

const a = tf.data.array([2, 4, 6]).repeat(5);
  
await a.forEachAsync(e => console.log(e*2));

输出：

4
8
12
4
8
12
4
8
12
4
8
12
4
8
12

参考：https://js.tensorflow.org/api/latest/#tf.data.Dataset.repeat