Función Tensorflow.js tf.layers.repeatVector()

Tensorflow.js es una biblioteca de código abierto desarrollada por Google para ejecutar modelos de aprendizaje automático y redes neuronales de aprendizaje profundo en el entorno del navegador o del Node.

La función tf.layers.repeatVector() se usa para repetir la entrada n veces en una nueva dimensión especificada. Es una función incorporada de la biblioteca TensorFlow’s.js.

Sintaxis:

tf.layers.repeatVector(n)

Parámetros:

• n: Entero, especificando el número de veces que se repetirá la entrada.

Valor devuelto: Devuelve el tf.layers.Layer

Ejemplo 1:

// 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)

Producción:

1, 5, 2

Ejemplo 2:

// 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();

Producción:

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

Referencia: https://js.tensorflow.org/api/latest/#layers.repeatVector