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

Tensorflow.js是由Google开发的开源库，用于在浏览器或节点环境中运行机器学习模型以及深度学习神经网络。

.toArray() 方法用于在数组中累积指定数据集的每个元素。

用法：

toArray()

参数：该方法不持有任何参数。

返回值：它返回 Promise( T[] )。

注意：

此方法仅适用于有限的数据集，即适合内存的数据集。
它用于测试目的，在可能的情况下应尽量避免。

范例1：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining dataset formed of an array
// of numbers
const res = tf.data.array([11, 12, 31, 43, 15, 64]);
  
// Calling toArray() method and
// Printing output
console.log(await res.toArray());

输出：

11, 12, 31, 43, 15, 64

范例2：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Calling toArray() method and
// Printing output
console.log(await tf.data.array([5.7, 
    -16, .31, 0, null, NaN]).toArray());

输出：

5.7, -16, 0.31, 0, , NaN

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

相关用法

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