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
Metody dziedziczone
wartość logiczna | równa się (Obiekt arg0) |
ostatnia klasa<?> | pobierzKlasę () |
wew | hashCode () |
ostateczna pustka | powiadomić () |
ostateczna pustka | powiadom wszystkich () |
Smyczkowy | doString () |
ostateczna pustka | czekaj (długi arg0, int arg1) |
ostateczna pustka | czekaj (długi arg0) |
ostateczna pustka | Czekać () |
abstrakcyjne środowisko wykonania | środowisko () Zwróć środowisko wykonawcze, w którym utworzono tę operację. |
abstrakcyjna operacja |
streszczenie Wyjście <V rozszerza TNumer > | jako wyjście () Zwraca symboliczny uchwyt tensora. |
streszczenie V rozszerza TNumber | asTensor () Zwraca tensor tego operandu. |
abstrakcyjny kształt | |
Klasa abstrakcyjna<V rozszerza TNumber > | typ () Zwraca typ tensora tego operandu |
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