Tensorflow.JS - TF.argmax
Em um tensor, o índice começa com 0.
Sintaxe
tf.argmax (tensor_input, eixo)Parâmetros
1. tensor_input é um tensor que tem elementos numéricos.
Pode ser 1 ou 2 dimensional.
2. Se o tensor for bidimensional, é possível especificar o eixo para obter um índice de valores máximos em linhas ou colunas.
Se eixo = 0, o índice dos valores máximos é retornado em coluna e se o eixo = 1, o índice de valores máximos é retornado em toda a linha.
Se o eixo não for especificado, ele retornará o índice de valores máximos em termos de coluna.
Retornar
Devolver um tensor com os índices de valor máximo.
Exemplo 1
Vamos criar um tensor unidimensional em JS que tenha valores inteiros e retorne um índice de valor máximo.
Tensorflow.JS - TF.argmax ()
Saída
90 é o máximo entre todos os elementos e está presente na 4ª posição. O índice é 3.
Então 3 é devolvido.
Exemplo 2
Vamos criar um tensor que tenha 2 dimensões com 4 linhas e 2 colunas que possuam valores inteiros e retornem índices de valor máximo nas colunas.
Tensorflow.JS - TF.argmax ()
Saída
Trabalhando
Tensor tensor [[10, 13], [15, 6], [67, 5], [10, 2]]
O valor máximo entre (10,15,67,10) é 67 e (13,6,5,2) é 13.
As posições do índice 67 e 13 são 2 e 0.
Exemplo 3
Vamos criar um tensor com 2 dimensões em JS com 4 linhas e 2 colunas com valores inteiros e retornar índices de valor máximo em linhas.
Tensorflow.JS - TF.argmax ()
Saída
Trabalhando
[[10, 13], [15, 6], [67, 5], [10, 2]]
Os valores máximos entre [10,13] é 13, [15, 6] é 15, [67, 5] é 67 e [10, 2] é 10.
Posições de índice de 13 é 1, 15 é 0, 67 é 0 e 10 é 0.
Exemplo 4
Vamos criar um tensor com 2 dimensões em JS com 4 linhas e 2 colunas com valores inteiros e retornar os índices dos valores máximos.
Tensorflow.JS - TF.argmax ()
Saída
Trabalhando
Tensor [[1,2], [3,4], [5,6], [7,8]]
O valor máximo na coluna [1,3,5,7] é 7 e seu índice é 3.
O valor máximo na coluna [2,4,6,8] é 8 e seu índice é 3.
Conclusão
Neste tensorflow.JS Tutorial, vimos como devolver o índice de elementos máximos presentes em um tensor usando o TF.Método argmax (). Em um tensor 2D, se o eixo = 0, o índice de valores máximos é retornado em colunas e se o eixo = 1, o índice de valores máximos é retornado em toda a linha. Por padrão, ele retornará o índice de valores máximos em termos de coluna.