Tensorflow.JS - TF.argmin
Em um tensor, o índice começa com 0.
Sintaxe:
tf.argmin (tensor_input, eixo)
Parâmetro:
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ínimos em linhas ou colunas.
Se eixo = 0, o índice de valores mínimos é retornado em coluna e se o eixo = 1, o índice de valores mínimos será retornado em toda a linha.
Se o eixo não for especificado, ele retornará o índice de valores mínimos em termos de coluna.
Retornar
Devolver um tensor com os índices de valor mínimo.
Exemplo 1:
Vamos criar um tensor unidimensional em JS que tenha valores inteiros e retorne um índice de valor mínimo.
Tensorflow.JS - TF.argmin ()
Saída:
34 é o mínimo entre todos os elementos e está presente na 1ª posição. Um índice é 0.
Então 0 é 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ínimo nas colunas.
Tensorflow.JS - TF.argmin ()
Saída:
Trabalhando:
Tensor tensor [[10, 13], [15, 6], [67, 5], [10, 2]]
O valor mínimo entre (10,15,67,10) é 10 e (13,6,5,2) é 2.
As posições de índice de 10 e 2 são 0 e 3.
Exemplo 3:
Vamos criar um tensor que tenha 2 dimensões em JS com 4 linhas e 2 colunas que possuem valores inteiros e retornem índices de valor mínimo entre linhas.
Tensorflow.JS - TF.argmin ()
Saída:
Trabalhando:
[[10, 13], [15, 6], [67, 5], [10, 2]]
Os valores mínimos entre [10,13] é 10, [15, 6] é 6, [67, 5] é 5 e [10, 2] é 2.
As posições de índice dos valores mínimos acima são - 0,1,1,1.
Exemplo 4:
Vamos criar um tensor que tenha 2 dimensões em JS com 4 linhas e 2 colunas que possuem valores inteiros e retorne os índices dos valores mínimos.
Tensorflow.JS - TF.argmin ()
Saída:
Trabalhando:
Tensor [[1,2], [3,4], [5,6], [7,8]]
O valor mínimo na coluna [1,3,5,7] é 1 e seu índice é 0.
O valor mínimo na coluna [2,4,6,8] é 2 e seu índice é 0.
Conclusão
Neste tensorflow.JS Tutorial, vimos como devolver o índice de elementos mínimos presentes em um tensor usando o TF.Método argmin (). Em um tensor 2D, se o eixo = 0, o índice de valores mínimos é retornado em coluna e, se o eixo = 1, o índice de valores mínimos é retornado em toda a linha. Por padrão, ele retornará o índice de valores mínimos em termos de coluna.