Flujo tensor.JS - TF.argmax
En un tensor, el índice comienza con 0.
Sintaxis
TF.ArgMax (tensor_input, eje)Parámetros
1. tensor_input es un tensor que tiene elementos numéricos.
Puede ser 1 o 2 dimensional.
2. Si el tensor es bidimensional, entonces es posible especificar el eje para obtener un índice de valores máximos en filas o columnas.
Si axis = 0, el índice de valores máximos se devuelve a través de la columna, y si axis = 1, el índice de valores máximos se devuelve a través de la fila.
Si no se especifica el eje, devolverá el índice de valores máximos en cuanto a columna.
Devolver
Devolver un tensor con los índices de valor máximo.
Ejemplo 1
Creemos un tensor unidimensional en JS que tiene valores enteros y devuelve un índice de valor máximo.
Flujo tensor.JS - TF.argmax ()
Producción
90 es el máximo entre todos los elementos y está presente en la cuarta posición. El índice es 3.
Entonces 3 se devuelve.
Ejemplo 2
Creemos un tensor que tiene 2 dimensiones con 4 filas y 2 columnas que tiene valores enteros y devuelve los índices de valor máximo en las columnas.
Flujo tensor.JS - TF.argmax ()
Producción
Laboral
Tensor tensor [[10, 13], [15, 6], [67, 5], [10, 2]]
El valor máximo entre (10,15,67,10) es 67 y (13,6,5,2) es 13.
Las posiciones de índice 67 y 13 son 2 y 0.
Ejemplo 3
Creemos un tensor con 2 dimensiones en JS con 4 filas y 2 columnas con valores enteros y devuelvan los índices de valor máximo en las filas.
Flujo tensor.JS - TF.argmax ()
Producción
Laboral
[[10, 13], [15, 6], [67, 5], [10, 2]]
Los valores máximos entre [10,13] son 13, [15, 6] es 15, [67, 5] es 67 y [10, 2] es 10.
Las posiciones de índice de 13 son 1, 15 es 0, 67 es 0 y 10 es 0.
Ejemplo 4
Creemos un tensor con 2 dimensiones en JS con 4 filas y 2 columnas con valores enteros y devuelvan los índices de los valores máximos.
Flujo tensor.JS - TF.argmax ()
Producción
Laboral
Tensor [[1,2], [3,4], [5,6], [7,8]]
El valor máximo en la columna [1,3,5,7] es 7, y su índice es 3.
El valor máximo en la columna [2,4,6,8] es 8, y su índice es 3.
Conclusión
En este flujo tensor.Tutorial JS, hemos visto cómo devolver el índice de elementos máximos presentes en un tensor utilizando el TF.método argMax (). En un tensor 2D, si axis = 0, el índice de valores máximos se devuelve a través de la columna, y si axis = 1, el índice de valores máximos se devuelve a través de la fila en cuanto a la fila. Por defecto, devolverá el índice de valores máximos en cuanto a columna.