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”
Javascript
// 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”
Javascript
// 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
Publicación traducida automáticamente
Artículo escrito por abhinavjain194 y traducido por Barcelona Geeks. The original can be accessed here. Licence: CCBY-SA