Tensorflow.js tf.data.Dataset.take()用法及代碼示例

Tensorflow.js是由Google開發的開源庫，用於在瀏覽器或節點環境中運行機器學習模型以及深度學習神經網絡。

.take() 方法用於生成最大的數據集數數所述數據集中的最重要的項目。

用法:

take(count)

Parameters:

count:用於創建不同數據集的是所述數據集中存在的元素數量。此外，如果計數未定義或為負數或高於指定數據集的維度，則新創建的數據集將保存給定數據集的每個項目。

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

範例1：

Javascript


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

輸出：

範例2：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Calling forEachAsync(), take() method 
// and printing output
await tf.data.array([31.1, 81.2, 5.1, 0, NaN, 'a']).
take(10.7).forEachAsync(op => console.log(op));

輸出：

31.1
81.2
5.1
0
NaN
a

參考： https://js.tensorflow.org/api/latest/#tf.data.Dataset.take

相關用法

注：本文由純淨天空篩選整理自nidhi1352singh大神的英文原創作品 Tensorflow.js tf.data.Dataset class .take() Method。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。