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.
La función tf.gatherND() se usa para recopilar cortes de la forma especificada por índices con el tensor de entrada.
Sintaxis:
tf.gatherND(tensor1, tensor2)
Parámetros: Esta función acepta los siguientes dos parámetros:
- tensor1: Es aquel tensor del que extraemos valores.
- tensor2: Es un tensor entero K-dimensional. Actúa como índices. Debe ser de tipo int32.
Valor devuelto: Devuelve el objeto tf.Tensor .
Ejemplo 1:
Javascript
// Importing the tensorflow.js library import * as tf from "@tensorflow/tfjs" // Initializing a 3d tensor as indices. let geeks = tf.tensor3d([1, 2, 3, 4], [2, 2, 1], 'int32'); // Initializing a 3d tensor as input. let input = tf.tensor1d([1, 2, 3]); // Gathering both input and indices. let answer = tf.gatherND(input, geeks); answer.print();
Producción:
Tensor [[2, 3], [3, 3]]
Ejemplo 2:
Javascript
// Importing the tensorflow.js library import * as tf from "@tensorflow/tfjs" // Gathering the tensor and indices together. let geeks = tf.gatherND(tf.tensor4d([[[[1], [2]], [[3], [4]]]]), tf.tensor2d([1, 1, 1, 0], [2,2], 'int32')); // Printing the final result. geeks.print();
Producción:
Tensor [[[4], [4]], [[4], [4]]]
Referencia: https://js.tensorflow.org/api/latest/#gatherND
Publicación traducida automáticamente
Artículo escrito por thacker_shahid y traducido por Barcelona Geeks. The original can be accessed here. Licence: CCBY-SA