Tensorflow.js es una biblioteca de código abierto desarrollada por Google para ejecutar modelos de aprendizaje automático, así como redes neuronales de aprendizaje profundo en el entorno del navegador o del Node.
La función .expandDims() se usa para encontrar el objeto tf.Tensor cuyo rango se expande con la ayuda de la forma del tensor insertando una dimensión en él.
Sintaxis:
tf.expandDims(x, axis?)
Parámetros:
- x: Es la entrada de tensor indicada cuyo tamaño se va a ampliar, y puede ser de tipo tf.Tensor, TypedArray o Array.
- eje: es el índice de tamaño establecido en el que se insertará la forma de 1. El valor por defecto es cero, es decir, la primera dimensión, y es de tipo número.
Valor devuelto: Devuelve el objeto tf.Tensor.
Ejemplo 1:
Javascript
// Importing the tensorflow.js library import * as tf from "@tensorflow/tfjs" // Defining tensor input and axis const y = tf.tensor1d([5, 6, 7, 8]); const axs = 1; // Calling tf.expandDims() method and // Printing output y.expandDims(axs).print();
Producción:
Tensor [[5], [6], [7], [8]]
Ejemplo 2:
Javascript
// Importing the tensorflow.js library import * as tf from "@tensorflow/tfjs" // Calling tf.expandDims() method and // Printing output tf.expandDims(tf.tensor1d([1.2, 3.5, 7.6, 9.7]), 1).print();
Producción:
Tensor [[1.2 ], [3.5 ], [7.5999999], [9.6999998]]
Referencia: https://js.tensorflow.org/api/latest/#expandDims
Publicación traducida automáticamente
Artículo escrito por nidhi1352singh y traducido por Barcelona Geeks. The original can be accessed here. Licence: CCBY-SA