Zwraca indeks o największej wartości spośród wszystkich wymiarów tensora.
Należy pamiętać, że w przypadku remisów tożsamość zwracanej wartości nie jest gwarantowana.
Stosowanie:
import tensorflow as tf
a = [1, 10, 26.9, 2.8, 166.32, 62.3]
b = tf.math.argmax(input = a)
c = tf.keras.backend.eval(b)
# c = 4
# here a[4] = 166.32 which is the largest element of a across axis 0
Stałe
Smyczkowy | OP_NAME | Nazwa tej operacji znana silnikowi rdzenia TensorFlow |
Metody publiczne
Wyjście <V> | jako wyjście () Zwraca symboliczny uchwyt tensora. |
statyczny <V rozszerza TNumber > ArgMax <V> | |
statyczny ArgMax < TInt64 > | |
Wyjście <V> | wyjście () |
Metody dziedziczone
Stałe
publiczny statyczny końcowy ciąg znaków OP_NAME
Nazwa tej operacji znana silnikowi rdzenia TensorFlow
Metody publiczne
publiczne wyjście <V> asOutput ()
Zwraca symboliczny uchwyt tensora.
Dane wejściowe operacji TensorFlow są wynikami innej operacji TensorFlow. Ta metoda służy do uzyskania symbolicznego uchwytu reprezentującego obliczenia danych wejściowych.
public static ArgMax <V> create (Zakres zakresu , Operand <? rozszerza TType > wejście, Operand <? rozszerza TNumber > wymiar, Klasa <V> typ wyjścia)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację ArgMax.
Parametry
zakres | aktualny zakres |
---|---|
wymiar | int32 lub int64, musi należeć do zakresu `[-rank(input), rank(input))`. Opisuje, który wymiar tensora wejściowego należy zmniejszyć. W przypadku wektorów użyj wymiaru = 0. |
Powroty
- nowa instancja ArgMax
public static ArgMax < TInt64 > create (Zakres zakresu , Operand <? rozszerza TType > wejście, Operand <? rozszerza TNumber > wymiar)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację ArgMax przy użyciu domyślnych typów wyjściowych.
Parametry
zakres | aktualny zakres |
---|---|
wymiar | int32 lub int64, musi należeć do zakresu `[-rank(input), rank(input))`. Opisuje, który wymiar tensora wejściowego należy zmniejszyć. W przypadku wektorów użyj wymiaru = 0. |
Powroty
- nowa instancja ArgMax