Tensorflow.JS - TF.Argmax
I en tensor starter indeksen med 0.
Syntaks
tf.Argmax (Tensor_input, Axis)Parametere
1. tensor_input er en tensor som har numeriske elementer.
Det kan være 1 eller 2 dimensjonal.
2. Hvis tensoren er todimensjonal, er det mulig å spesifisere aksen for å få en indeks med maksimale verdier i rader eller kolonner.
Hvis Axis = 0, returneres indeksen for maksimumsverdier på tvers av kolonnemessig, og hvis Axis = 1, returneres indeksen for maksimumsverdier over radmessig.
Hvis aksen ikke er spesifisert, vil den returnere indeksen for maksimumsverdier kolonnemessig.
Komme tilbake
Returner en tensor med maksimale verdiindekser.
Eksempel 1
La oss opprette en enimensjonal tensor i JS som har heltallverdier og returnerer en maksimal verdiindeks.
Tensorflow.JS - TF.Argmax ()
Produksjon
90 er det maksimale blant alle elementene og er til stede i 4. plassering. Indeks er 3.
Så 3 blir returnert.
Eksempel 2
La oss lage en tensor som har to dimensjoner med 4 rader og 2 kolonner som har heltallverdier og returnerer maksimale verdiindekser på tvers av kolonner.
Tensorflow.JS - TF.Argmax ()
Produksjon
Arbeider
Tensor Tensor [[10, 13], [15, 6], [67, 5], [10, 2]]
Maksimal verdi blant (10,15,67,10) er 67 og (13,6,5,2) er 13.
Indeksposisjoner 67 og 13 er 2 og 0.
Eksempel 3
La oss lage en tensor med to dimensjoner i JS med 4 rader og 2 kolonner med heltallverdier og returnere maksimale verdiindekser på tvers av rader.
Tensorflow.JS - TF.Argmax ()
Produksjon
Arbeider
[[10, 13], [15, 6], [67, 5], [10, 2]]
Maksimumsverdier blant [10,13] er 13, [15, 6] er 15, [67, 5] er 67 og [10, 2] er 10.
Indeksposisjoner på 13 er 1, 15 er 0, 67 er 0 og 10 er 0.
Eksempel 4
La oss lage en tensor med to dimensjoner i JS med 4 rader og 2 kolonner med heltallverdier og returnere indeksene for maksimumsverdiene.
Tensorflow.JS - TF.Argmax ()
Produksjon
Arbeider
Tensor [[1,2], [3,4], [5,6], [7,8]]
Maksimal verdi i kolonnen [1,3,5,7] er 7, og indeksen er 3.
Maksimal verdi i kolonnen [2,4,6,8] er 8, og indeksen er 3.
Konklusjon
I denne tensorflow.JS -opplæring, vi har sett hvordan vi kan returnere indeksen over maksimale elementer som er til stede i en tensor ved hjelp av TF.ArgMax () -metode. I en 2D-tensor, hvis Axis = 0, returneres indeksen for maksimale verdier på tvers av kolonnemessig, og hvis Axis = 1, returneres indeksen for maksimumsverdier over radmessig. Som standard vil den returnere indeksen over maksimumsverdier kolonne-messig.