Método Tensorflow.js tf.layers computeOutputShape()

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:

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

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