Tensorflow.JS - TF.Argmax

„TF.argmax () im Tensorflow.JS gibt den Index des maximalen Elements aus dem Elementsatz in einem Tensor zurück. Wenn der Tensor zweidimensional ist, kann es möglich sein, einen Index mit maximalen Werten über Zeilen und Spalten zurückzugeben.”

In einem Tensor beginnt der Index mit 0.

Syntax

tf.Argmax (Tensor_input, Achse)

Parameter

1. Tensor_input ist ein Tensor mit numerischen Elementen.

Es kann 1OR 2 dimensional sein.

2. Wenn der Tensor zweidimensional ist, ist es möglich, die Achse anzugeben, um einen Index von maximalen Werten in Zeilen oder Spalten zu erhalten.

Wenn Axis = 0 ist, wird der Index der maximalen Werte über die Spaltenwise zurückgegeben, und wenn Axis = 1, wird der Index der maximalen Werte über zeilenweise über die Zeile zurückgegeben.

Wenn die Achse nicht angegeben ist, gibt sie den Spaltenindex der maximalen Werte zurück.

Zurückkehren
Geben Sie einen Tensor mit den Maximalwertindizes zurück.

Beispiel 1
Erstellen wir einen eindimensionalen Tensor in JS mit ganzzahligen Werten und geben einen Maximalwertindex zurück.

Tensorflow.JS - TF.argmax ()

Ausgang

90 ist das Maximum unter allen Elementen und ist in der 4. Position vorhanden. Index ist 3.

Also wird 3 zurückgegeben.

Beispiel 2
Erstellen wir einen Tensor mit 2 Dimensionen mit 4 Zeilen und 2 Spalten mit ganzzahligen Werten und geben maximale Wertindizes über Spalten hinweg zurück.

Tensorflow.JS - TF.argmax ()

Ausgang

Arbeiten
Tensor Tensor [[10, 13], [15, 6], [67, 5], [10, 2]]
Der Höchstwert unter (10,15,67,10) beträgt 67 und (13,6,5,2) 13.
Indexpositionen 67 und 13 sind 2 und 0.

Beispiel 3
Erstellen wir einen Tensor mit 2 Dimensionen in JS mit 4 Zeilen und 2 Spalten mit Ganzzahlwerten und geben maximale Wertindizes über Zeilen zurück.

Tensorflow.JS - TF.argmax ()

Ausgang

Arbeiten
[[10, 13], [15, 6], [67, 5], [10, 2]]
Die maximalen Werte unter [10,13] beträgt 13, [15, 6] 15, [67, 5] ist 67 und [10, 2] 10 IS 10.
Indexpositionen von 13 sind 1, 15 ist 0, 67 ist 0 und 10 Is 0.

Beispiel 4
Erstellen wir einen Tensor mit 2 Dimensionen in JS mit 4 Zeilen und 2 Spalten mit Ganzzahlwerten und geben die Indizes der maximalen Werte zurück.

Tensorflow.JS - TF.argmax ()

Ausgang

Arbeiten
Tensor [[1,2], [3,4], [5,6], [7,8]]
Der Maximalwert in der Spalte [1,3,5,7] beträgt 7 und sein Index beträgt 3.
Der Maximalwert in der Spalte [2,4,6,8] beträgt 8 und sein Index beträgt 3.

Abschluss

In diesem Tensorflow.JS Tutorial haben wir gesehen.argmax () Methode. In einem 2D-Tensor wird bei Axis = 0 der Index der maximalen Werte über Säulen hinweg zurückgegeben, und wenn Axis = 1, wird der Index der maximalen Werte über die Zeile hinweg zurückgegeben. Standardmäßig gibt es den Spaltenindex der maximalen Werte zurück.