Función Tensorflow.js tf.unique()

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

Publicación traducida automáticamente

Artículo escrito por sweetyty y traducido por Barcelona Geeks. The original can be accessed here. Licence: CCBY-SA

Deja una respuesta

Tu dirección de correo electrónico no será publicada. Los campos obligatorios están marcados con *