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

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

tf.data.Dataset.forEachAsync() 函数在对数据集的每个元素应用一个函数后使用

用法：

forEachAsync(f)

参数：

• f:应用于每个数据集元素的函数。

返回值:它返回 promise<void>

范例1：

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

输出：

1
2
3

范例2：

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

输出：

2
4
6

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