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:
Javascript
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.
Javascript
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
Publicación traducida automáticamente
Artículo escrito por parasmadan15 y traducido por Barcelona Geeks. The original can be accessed here. Licence: CCBY-SA