Tensorflow.js es una biblioteca de código abierto desarrollada por Google para ejecutar modelos de aprendizaje automático, así como redes neuronales de aprendizaje profundo en el entorno del navegador o del Node.
La función .computeOutputShape() se usa para enumerar la forma de salida de la capa indicada. Y supone que la capa se creará para que coincida con la forma de entrada suministrada.
Sintaxis:
computeOutputShape(inputShape)
Parámetros:
- inputShape: Es la forma establecida, es decir, un conjunto de números enteros o una lista de conjuntos de formas. Además, las tuplas Shape pueden contener vacíos a favor de tamaños libres, en lugar de un número entero. Puede ser de tipo ((nulo | número)[]|(nulo | número)[][]).
Valor devuelto: Devuelve (nulo | número)[]|(nulo | número)[][].
Ejemplo 1:
Javascript
// Importing the tensorflow.js library import * as tf from "@tensorflow/tfjs" // Creating a model const model = tf.sequential(); // Adding a layer model.add(tf.layers.dense({units: 1, inputShape: [3]})); // Defining inputShape const inputShape = [6, 2, 6]; // Calling computeOutputShape() method with its // parameter const val = model.layers[0].computeOutputShape(inputShape); // Printing output console.log(val);
Producción:
6,2,1
Ejemplo 2:
Javascript
// Importing the tensorflow.js library import * as tf from "@tensorflow/tfjs" // Creating a model const model = tf.sequential(); // Adding layers model.add(tf.layers.dense({units: 1, inputShape: [3]})); model.add(tf.layers.dense({units: 5})); // Defining inputShape const inputShape1 = [6, 2, 6, null]; const inputShape2 = [6.5, 2.6, 9.1, NaN]; // Calling computeOutputShape() method with its // parameter const val1 = model.layers[0].computeOutputShape(inputShape1); const val2 = model.layers[1].computeOutputShape(inputShape2); // Printing output console.log(val1); console.log(val2);
Producción:
6,2,6,1 6.5,2.6,9.1,5
Referencia: https://js.tensorflow.org/api/latest/#tf.layers.Layer.computeOutputShape
Publicación traducida automáticamente
Artículo escrito por nidhi1352singh y traducido por Barcelona Geeks. The original can be accessed here. Licence: CCBY-SA