Función Tensorflow.js tf.reshape()

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. También ayuda a los desarrolladores a desarrollar modelos ML en lenguaje JavaScript y puede usar ML directamente en el navegador o Node.js.

La función tf.reshape() se usa para remodelar un tensor dado con la forma especificada.

Sintaxis:

tf.reshape(x, shape)

Parámetros: Esta función tiene los siguientes parámetros:

• x: Es el tensor de entrada al que hay que dar forma.
• forma: necesitamos pasar una array de números para definir la forma de salida.

Valor devuelto: Devuelve un tf.Tensor.

Ejemplo 1:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
const x = tf.tensor1d([10, 15, 16, 24]);
  
// Print the tensor
x.reshape([2, 2]).print();

Producción:

Tensor
    [[10, 15],
     [16, 24]]

Ejemplo 2:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Using 2d
const x = tf.tensor2d(
  [1, 2, 3, 4, 5, 6, 7, 8, 9], [3, 3]
);
x.reshape([3, 3]).print();
  
// Using 3d
const y = tf.tensor3d(
  [[[1], [2]], [[3], [4]]]
);
y.reshape([2, 2]).print();
  
// Using 4d
const z = tf.tensor4d(
  [11, 12, 13, 14], [1, 2, 2, 1]
);
z.reshape([2, 2]).print();

Producción:

Tensor
    [[1, 2, 3],
     [4, 5, 6],
     [7, 8, 9]]
Tensor
    [[1, 2],
     [3, 4]]
Tensor
    [[11, 12],
     [13, 14]]

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