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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。