Función Tensorflow.js tf.spaceToBatchND()

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.spaceToBatchND() se usa para dividir las dimensiones espaciales del espacio de entrada especificado en una array de bloques que tienen la forma blockShape, donde blockshape es el parámetro. Aquí, las dimensiones espaciales se rellenan de acuerdo con la array de relleno especificada. 

Sintaxis:

tf.spaceToBatchND (x, blockShape, paddings)

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

• x: un tensor especificado de dimensión N con forma de «[lote] + forma espacial + forma restante», donde forma espacial tiene dimensiones M.
• blockShape: es una array 1-D que debe tener la forma de [M], ya que todos los valores deben ser mayores o iguales a 1.
• paddings: Una array 2-D que tiene forma de [M, 2], para todos los valores debe ser mayor o igual a 0. Aquí padding es igual a [padStart, padEnd].

Valor devuelto: Devuelve un tensor de la versión dividida del espacio de entrada especificado.

Ejemplo 1:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
 
// Initializing a Tensor
const x = tf.tensor4d(
    [5, 10, 15, 20, 25, 30],
    [1, 3, 2, 1]
);
 
// Initializing blockShape and paddings parameters
const blockShape = [1, 1];
const paddings = [[0, 0], [0, 0]];
 
// Calling the .spaceToBatchND() function over
// the above parameters and Tensor
x.spaceToBatchND(blockShape, paddings).print();

Producción: 

Tensor
   [[[[5 ],
      [10]],

     [[15],
      [20]],

     [[25],
      [30]]]]

Ejemplo 2:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
 
// Initializing a Tensor
const x = tf.tensor4d([2, 4, 6, 8], [1, 2, 2, 1]);
 
// Using the blockShape and paddings as the
// parameter for the .spaceToBatchND() function
x.spaceToBatchND([2, 2], [[0, 0], [0, 0]]).print();

Producción: 

Tensor
   [ [ [[2],]],

     [ [[4],]],

     [ [[6],]],

     [ [[8],]]]

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