Tensorflow.JS - TF.argmax
In un tensore, l'indice inizia con 0.
Sintassi
tf.argmax (tensor_input, asse)Parametri
1. tensor_input è un tensore che ha elementi numerici.
Può essere 1o 2 dimensionale.
2. Se il tensore è bidimensionale, è possibile specificare l'asse per ottenere un indice dei valori massimi in righe o colonne.
Se Asse = 0, l'indice dei valori massimi viene restituito attraverso la colonna e se Asse = 1, l'indice dei valori massimi viene restituito attraverso la riga dal punto di vista della riga.
Se l'asse non è specificato, restituirà l'indice dei valori massimi per quanto riguarda la colonna.
Ritorno
Restituire un tensore con gli indici di valore massimo.
Esempio 1
Creiamo un tensore monodimensionale in JS che ha valori interi e restituiamo un indice di valore massimo.
Tensorflow.JS - TF.argmax ()
Produzione
90 è il massimo tra tutti gli elementi ed è presente in quarta posizione. L'indice è 3.
Quindi 3 viene restituito.
Esempio 2
Creiamo un tensore che ha 2 dimensioni con 4 righe e 2 colonne che hanno valori interi e restituiscono indici di valore massimo tra le colonne.
Tensorflow.JS - TF.argmax ()
Produzione
Lavorando
Tensor tensore [[10, 13], [15, 6], [67, 5], [10, 2]]
Il valore massimo tra (10,15,67,10) è 67 e (13,6,5,2) è 13.
Le posizioni dell'indice 67 e 13 sono 2 e 0.
Esempio 3
Creiamo un tensore con 2 dimensioni in JS con 4 righe e 2 colonne con valori interi e restituire gli indici di valore massimo attraverso le righe.
Tensorflow.JS - TF.argmax ()
Produzione
Lavorando
[[10, 13], [15, 6], [67, 5], [10, 2]]
I valori massimi tra [10,13] sono 13, [15, 6] è 15, [67, 5] è 67 e [10, 2] è 10.
Le posizioni dell'indice di 13 sono 1, 15 è 0, 67 è 0 e 10 è 0.
Esempio 4
Creiamo un tensore con 2 dimensioni in JS con 4 righe e 2 colonne con valori interi e restituire gli indici dei valori massimi.
Tensorflow.JS - TF.argmax ()
Produzione
Lavorando
Tensor [[1,2], [3,4], [5,6], [7,8]]
Il valore massimo nella colonna [1,3,5,7] è 7 e il suo indice è 3.
Il valore massimo nella colonna [2,4,6,8] è 8 e il suo indice è 3.
Conclusione
In questo tensorflow.JS Tutorial, abbiamo visto come restituire l'indice degli elementi massimi presenti in un tensore usando il TF.Metodo argmax (). In un tensore 2D, se AXIS = 0, l'indice dei valori massimi viene restituito attraverso la colonna e se l'asse = 1, l'indice dei valori massimi viene restituito attraverso la riga per quanto riguarda la riga. Per impostazione predefinita, restituirà l'indice dei valori massimi per quanto riguarda la colonna.