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

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

tf.data.Dataset.forEachAsync() 函数在将函数应用于数据集的每个元素后使用。我们可以使用此方法打印数组、修改数组的值等。

用法:

forEachAsync(f)

参数：

f:它接受一个函数，该函数应用于数据集的每个元素

返回值：它返回一个承诺

范例1：

Javascript


// Creating an array
const arr = tf.data.array([10, 20, 30, 40, 50]);
  
// Applying the function on the array
await arr.forEachAsync(element => console.log(element));

输出：

范例2：在这个例子中，我们将传递一个计算元素平方的函数。

Javascript


// Creating an array
const arr = tf.data.array([1, 2, 3, 4, 5]);
  
// Applying the function on the array
await arr.forEachAsync(element => console.log(element*element));

输出：

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

相关用法

注：本文由纯净天空筛选整理自parasmadan15大神的英文原创作品 Tensorflow.js tf.data.Dataset class .forEachAsync() Method。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。