Función Tensorflow.js tf.batchToSpaceND()

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.

La función tf.batchToSpaceND() se usa para reestructurar el «lote» dado de su dimensión cero a dimensiones «M+1» que tienen la forma de «forma de bloque + [lote]», donde la forma de bloque es el parámetro y el lote es el tensor especificado. Aquí, el recorte se realiza en el resultado intermedio según la array de recorte dada. 

tf.batchToSpaceND (x, blockShape, crops)

Parámetros: Esta función acepta tres parámetros que se ilustran a continuación:

• x: un tensor de lote especificado de dimensión N con una forma de «[lote] + forma espacial + forma restante», donde forma espacial tiene dimensiones M.
• blockShape: una array 1-D debe tener la forma de [M], ya que todos los valores deben ser mayores o iguales a 1.
• cultivos: una forma de array 2-D de [M, 2] para la cual todos los valores deben ser mayores o iguales a 0. Aquí cultivos [i] = [cropStart, cropEnd] que define la porción para recortar desde la dimensión de entrada i + 1.

Valor de retorno: Devuelve un tensor de la versión remodelada del lote especificado.

Ejemplo 1: 

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
 
// Initializing a 3D Tensor of batch to reshape
const x = tf.tensor3d([5, 10, 15, 20, 25], [5, 1, 1]);
 
// Initializing blockShape and crops parameter
const blockShape = [1, 1];
const crops = [[0, 0], [0, 0]];
 
// Calling the .batchToSpaceND() function over
// the above parameters and Tensor
x.batchToSpaceND(blockShape, crops).print();

Producción:

Tensor
    [ [[5 ],],

      [[10],],

      [[15],],

      [[20],],

      [[25],]]

Ejemplo 2:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
 
// Initializing a 4D Tensor of batch to restructure
const x = tf.tensor4d([0, 2, 4, 6, 8, 10], [6, 1, 1, 1]);
 
// Using the blockShape and crops as the parameter
// for the .batchToSpaceND() function
x.batchToSpaceND([3, 2], [[0, 0], [0, 0]]).print();

Producción:

Tensor
    [[[[0 ],
       [2 ]],

      [[4 ],
       [6 ]],

      [[8 ],
       [10]]]]

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