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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。