Tensorflow.JS - TF.Argmax
W tensorze indeks zaczyna się od 0.
Składnia
tf.argmax (tensor_input, oś)Parametry
1. tensor_input to tensor, który ma elementy numeryczne.
Może być 1 lub 2 wymiarowy.
2. Jeśli tensor jest dwuwymiarowy, możliwe jest określenie osi, aby uzyskać wskaźnik maksymalnych wartości w wierszach lub kolumnach.
Jeśli oś = 0, wskaźnik wartości maksymalnych jest zwracany w stosunku do kolumny, a jeśli oś = 1, wskaźnik wartości maksymalnych jest zwracany przez rzędowy.
Jeśli oś nie zostanie określona, zwróci indeks maksymalnych wartości kolumnowych.
Powrót
Zwróć tensor z wskaźnikami maksymalnej wartości.
Przykład 1
Utwórzmy jeden wymiarowy tensor w JS, który ma wartości całkowitej i zwraca wskaźnik wartości maksymalnej wartości.
Tensorflow.JS - TF.argmax ()
Wyjście
90 jest maksimum spośród wszystkich elementów i jest obecny w czwartej pozycji. Indeks to 3.
Więc 3 zostaje zwrócone.
Przykład 2
Utwórzmy tensor, który ma 2 wymiary z 4 rzędami i 2 kolumnami, które mają wartości liczb całkowitych i zwraca wskaźniki wartości maksymalnej w kolumnach.
Tensorflow.JS - TF.argmax ()
Wyjście
Pracujący
Tensor tensorowy [[10, 13], [15, 6], [67, 5], [10, 2]]
Maksymalna wartość wśród (10,15 67,10) wynosi 67, a (13,6,5,2) wynosi 13.
Pozycje indeksu 67 i 13 to 2 i 0.
Przykład 3
Utwórzmy tensor z 2 wymiarami w JS z 4 wierszami i 2 kolumnami o wartościach całkowitych i zwróć maksymalną wartość wskaźnik.
Tensorflow.JS - TF.argmax ()
Wyjście
Pracujący
[[10, 13], [15, 6], [67, 5], [10, 2]]
Maksymalne wartości wśród [10,13] to 13, [15, 6] to 15, [67, 5] to 67, a [10, 2] to 10.
Pozycje indeksu 13 to 1, 15 to 0, 67 to 0 i 10 to 0.
Przykład 4
Utwórzmy tensor z 2 wymiarami w JS z 4 wierszami i 2 kolumnami o wartościach całkowitych i zwróć wskaźniki wartości maksymalnych.
Tensorflow.JS - TF.argmax ()
Wyjście
Pracujący
Tensor [[1,2], [3,4], [5,6], [7,8]]
Maksymalna wartość w kolumnie [1,3,5,7] wynosi 7, a jej wskaźnik to 3.
Maksymalna wartość w kolumnie [2,4,6,8] wynosi 8, a jej wskaźnik to 3.
Wniosek
W tym tensorflow.Samouczek JS, widzieliśmy, jak zwrócić wskaźnik maksymalnych elementów obecnych w tensorze za pomocą TF.metoda argmax (). W tensorze 2D, jeśli oś = 0, wskaźnik wartości maksymalnych jest zwracany w stosunku do kolumny, a jeśli oś = 1, wskaźnik wartości maksymalnych jest zwracany przez rzędowy. Domyślnie zwróci indeks maksymalnych wartości kolumnowych.