Función Tensorflow.js tf.util.sizeFromShape()

Tensorflow.js es una biblioteca de código abierto que está desarrollando 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 .util.sizeFromShape() se usa para encontrar el número de elementos, es decir, el tamaño de la entrada de tensor indicada de la forma indicada.

Sintaxis:  

tf.util.sizeFromShape(shape)

Parámetros:  

• forma: Es la forma indicada cuyo tamaño se va a devolver. Es de tipo número[].

Valor devuelto: Devuelve un número.

Ejemplo 1:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
 
// Defining shape
const sh = [5, 8, 1];
 
// Calling tf.util.sizeFromShape() method
const num = tf.util.sizeFromShape(sh);
 
// printing output
console.log(num);

Producción:

40

Ejemplo 2:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
 
// Calling tf.util.sizeFromShape() method and
// printing output
console.log(tf.util.sizeFromShape([1.4, 5.6, 9.0]));

Producción:

70.55999999999999

Referencia: https://js.tensorflow.org/api/1.0.0/#sizeFromShape