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

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

tf.data.Dataset 類 .prefetch() 函數用於生成一個數據集，該數據集從該給定數據集中預取指定元素。

用法:

prefetch (bufferSize)

參數：此函數接受如下所示的參數：

bufferSize:它是一個整數值，指定要預取的元素數。

返回值：它返回一個元素數據集。

範例1：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
 
// Calling the .prefetch() function over
// the specified dataset of some elements
const a = tf.data.array([5, 10, 15, 20]).prefetch(4);
 
// Getting the dataset of prefetched elements
await a.forEachAsync(a => console.log(a));

輸出：

範例2：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
 
// Specifying a dataset of some elements
const a = tf.data.array(["a", "b", "c", "d", "e"]);
 
// Calling the .prefetch() function over
// the above dataset along with the
// batch of size 2
const b = a.batch(2)
const c = b.prefetch(2)
 
// Getting the dataset of prefetched elements
await c.forEachAsync(c => console.log(c));

輸出：

Tensor
    ['a', 'b']
Tensor
    ['c', 'd']
Tensor
    ['e']

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

相關用法

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