Tensorflow.js es una biblioteca de código abierto que está desarrollando 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 .unique() se usa para encontrar elementos únicos a lo largo de un eje de un tensor.
Sintaxis:
tf.unique (x, axis?)
Parámetros:
- x: Es una primera entrada de tensor que puede ser de tipo tf.Tensor, o TypedArray, o Array. Los parámetros pueden ser de estos tipos (int32, string, bool).
- eje ( número ): es una segunda entrada de tensor que es opcional. El eje del tensor se utiliza para encontrar los elementos únicos.
Valor devuelto: Devuelve los valores e índices.
Ejemplo 1: En este ejemplo, estamos definiendo un tensor de entrada de tipo entero y luego imprimiendo su valor único. Para crear un tensor de entrada, utilizamos el método .tensor1d() y, para imprimir la salida, utilizamos el método .print () .
Javascript
// Importing the tensorflow.js library import * as tf from "@tensorflow/tfjs" // Defining tensor input elements const a = tf.tensor1d([1, 1, 2, 2, 4, 4, 7, 7, 8]); // Calling unique() method const {values, indices} = tf.unique(a); // Printing the values and indices values.print(); // [1, 2, 4, 7, 8,] indices.print(); // [0, 0, 1, 1, 2, 2, 3, 3, 4]
Producción:
Tensor [1, 2, 4, 7, 8] Tensor [0, 0, 1, 1, 2, 2, 3, 3, 4]
Ejemplo 2: En este ejemplo se ha utilizado el parámetro del eje. Para crear un tensor de entrada, estamos utilizando el método .tensor2d() y para imprimir la salida estamos usando el método .print() .
Javascript
// Importing the tensorflow.js library import * as tf from "@tensorflow/tfjs" // Defining tensor input elements const a = tf.tensor2d([[0, 0, 0], [2, 0, 0], [2, 0, 0]]); // Calling the unique() method const {values, indices} = tf.unique(a, 0) // Printing the values and indices values.print(); // [[1, 0, 0], // [2, 0, 0]] indices.print(); // [0, 0, 1]
Producción:
Tensor [[0, 0, 0], [2, 0, 0]] Tensor [0, 1, 1]
Referencia: https://js.tensorflow.org/api/latest/#unique