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

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

.data.array() 方法用于基于由元素组成的数组形成数据集。

用法：

tf.data.array(items)

Parameters:

items:它是由要在数据集(如项目)中解析的元素组成的声明数组，它可以是 tf.void、数字、字符串、TypedArray、tf.Tensor、tf.Tensor[] 或 {[key: string]:tf.Tensor、数字或字符串}[]。

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

范例1：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining dataset formed of an array
// of objects and calling data.array()
const res = tf.data.array([
    {'element':5}, 
    {'element':6}, 
    {'element':7}
]);
  
// Calling forEachAsync() method and
// Printing output
await res.forEachAsync(op => console.log(op));

输出：

{
  "element":5
}
{
  "element":6
}
{
  "element":7
}

范例2：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining dataset formed of an array
// of numbers and calling data.array()
const res = tf.data.array([4.6, 7.9, 9.6, 2.6, 8.9]);
  
// Calling forEachAsync() method and
// Printing output
await res.forEachAsync(op => console.log(op));

输出：

4.6
7.9
9.6
2.6
8.9

参考： https://js.tensorflow.org/api/latest/#data.array

相关用法

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