Función Tensorflow.js tf.expandDims()

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 .expandDims() se usa para encontrar el objeto tf.Tensor cuyo rango se expande con la ayuda de la forma del tensor insertando una dimensión en él.

Sintaxis: 

tf.expandDims(x, axis?)

Parámetros:  

• x: Es la entrada de tensor indicada cuyo tamaño se va a ampliar, y puede ser de tipo tf.Tensor, TypedArray o Array.
• eje: es el índice de tamaño establecido en el que se insertará la forma de 1. El valor por defecto es cero, es decir, la primera dimensión, y es de tipo número.

Valor devuelto: Devuelve el objeto tf.Tensor.

Ejemplo 1:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining tensor input and axis
const y = tf.tensor1d([5, 6, 7, 8]);
const axs = 1;
  
// Calling tf.expandDims() method and
// Printing output
y.expandDims(axs).print();

Producción:

Tensor
    [[5],
     [6],
     [7],
     [8]]

Ejemplo 2:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Calling tf.expandDims() method and
// Printing output
tf.expandDims(tf.tensor1d([1.2, 3.5, 7.6, 9.7]), 1).print();

Producción:

Tensor
    [[1.2      ],
     [3.5      ],
     [7.5999999],
     [9.6999998]]

Referencia: https://js.tensorflow.org/api/latest/#expandDims