Función Tensorflow.js tf.topk()

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.

La función tf.topk() junto con la última dimensión también se usa para encontrar los valores e índices de las k entradas más grandes.

Sintaxis:

tf.topk (x, k?, sorted?)

Parámetros:

• x: 1-D o superior tf.Tensor con la última dimensión siendo al menos k.
• k: Es el número de elementos a buscar.
• sorted: Es el valor booleano. Si es cierto, los k elementos resultantes se ordenarán por los valores en orden descendente.

Valor devuelto: {valores: tf.Tensor, índices: tf.Tensor}. Devuelve un objeto con dos tensores que contienen los valores e índices.

Ejemplo 1:

const tf = require("@tensorflow/tfjs")
  
// Creating a 2d tensor
const a = tf.tensor2d([[1, 20, 3], [4, 3, 1], [8, 9, 10]]);
const {values, indices} = tf.topk(a);
  
// Printing the values and indices
values.print();
indices.print();

Producción:

Tensor
    [[20],
     [4 ],
     [10]]
Tensor
    [[1],
     [0],
     [2]]

Ejemplo 2: En este ejemplo, proporcionaremos el argumento k para obtener las k entradas más grandes.

const tf = require("@tensorflow/tfjs")
  
// Creating a 2d tensor
const a = tf.tensor2d([[1, 20, 3], [4, 3, 1], [8, 9, 10]]);
const {values, indices} = tf.topk(a, 3);
  
// Printing the values and indices
values.print();
indices.print();

Producción:

Como hemos pasado k = 3, obtenemos 3 valores más grandes en nuestro resultado.

Tensor
    [[20, 3, 1],
     [4 , 3, 1],
     [10, 9, 8]]
Tensor
    [[1, 2, 0],
     [0, 1, 2],
     [2, 1, 0]]

Referencia: https://js.tensorflow.org/api/latest/#topk