Tensorflow.js tf.layers.repeatVector()用法及代碼示例

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

tf.layers.repeatVector() 函數用於在新的指定維度中重複輸入 n 次。它是 TensorFlow's.js 庫的內置函數。

用法:

tf.layers.repeatVector(n)

參數：

n:整數，指定重複輸入的次數。

返回值：它返回 tf.layers.Layer

範例1：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
    
  
// Create a new model
const model = tf.sequential();
    
// Add repeatVector layer to the model
model.add(tf.layers.repeatVector({
    n:5, inputShape:[2]}
));
    
const x = tf.tensor2d([[10, 15]]);
   
console.log(model.predict(x).shape)

輸出：

1, 5, 2

範例2：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
    
// Create a new model
const model = tf.sequential();
    
// Add repeatVector layer to the model
model.add(tf.layers.repeatVector(
      {n:8, inputShape:[2]}
));
   
const x = tf.tensor2d([[0,1]]);
   
model.predict(x).print();

輸出：

Tensor
    [[[0, 1],
      [0, 1],
      [0, 1],
      [0, 1],
      [0, 1],
      [0, 1],
      [0, 1],
      [0, 1]]]

參考： https://js.tensorflow.org/api/latest/#layers.repeatVector

相關用法

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