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