Función Tensorflow.js tf. depthToSpace()

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 en Node.js.

El tf. depthToSpace() es una función incorporada de la biblioteca tensorflow.js , que se utiliza para reorganizar los datos en el tensor de entrada, donde los valores de la dimensión de profundidad se mueven en bloques espaciales a las dimensiones de altura y anchura. Reorganiza los datos de profundidad en bloques de datos espaciales.

Sintaxis:

tensor.depthToSpace(input, blocksize, dataformat)

Parámetros:

• entrada: el tensor dado
• blocksize: el ancho del tensor de salida es profundidad *blockSize.
• formato de datos : especifica el diseño del tensor dado y resultante. Tiene dos opciones: “NHWC”: [lote, alto, ancho, canales] y “NCHW”: [lote, canales, alto, ancho]

Valor devuelto: Devuelve el tensor reordenado del mismo tipo de dato.

Ejemplo 1: Uso del formato “NHWC”

// Importing the tensorflow.Js library
import * as tf from "@tensorflow/tfjs"
 
// Create a new tensor
var input = tf.tensor4d([1, 3, 5, 7], [1, 1, 1, 4]);
 
// define block size
var blockSize = 2;
 
// define data format
var dataFormat = "NHWC";
 
// rearrange data
var val = tf.depthToSpace(input, blockSize, dataFormat);
 
// print the tensor
val.print();

 Producción:

Tensor
    [[[[1],
       [3]],

      [[5],
       [7]]]]

Ejemplo 2: usando el formato “NCHW”

// Importing the tensorflow.Js library
import * as tf from "@tensorflow/tfjs"
 
// Create a new tensor
var input = tf.tensor4d([1, 3, 5, 7], [1, 4, 1, 1]);
 
// define block size
var blockSize = 2;
 
// define data format
var dataFormat = "NCHW";
 
// rearrange data
var tr = tf.depthToSpace(input, blockSize, dataFormat);
 
// print the tensor
tr.print();

 
Producción: 

Tensor
    [[[[1, 3],
       [5, 7]]]]

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