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:
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)
Producción:
1, 5, 2
Ejemplo 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();
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