Возвращает индекс с наибольшим значением по измерениям тензора.
Обратите внимание, что в случае связей идентичность возвращаемого значения не гарантируется.
Использование:
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
Константы
Нить | OP_NAME | Название этой операции, известное основному движку TensorFlow. |
Публичные методы
Выход <В> | какВывод () Возвращает символический дескриптор тензора. |
static <V расширяет TNumber > ArgMax <V> | |
статический ArgMax <TInt64> | |
Выход <В> | выход () |
Унаследованные методы
Константы
общедоступная статическая финальная строка OP_NAME
Название этой операции, известное основному движку TensorFlow.
Публичные методы
общедоступный вывод <V> asOutput ()
Возвращает символический дескриптор тензора.
Входные данные для операций TensorFlow являются выходными данными другой операции TensorFlow. Этот метод используется для получения символического дескриптора, который представляет собой вычисление входных данных.
public static ArgMax <V> create (область действия , операнд <? расширяет TType > ввод, операнд <? расширяет измерение TNumber >, Class<V> выходной тип)
Фабричный метод для создания класса, обертывающего новую операцию ArgMax.
Параметры
объем | текущий объем |
---|---|
измерение | int32 или int64 должен находиться в диапазоне `[-rank(input), Rank(input))`. Описывает, какое измерение входного тензора нужно уменьшить. Для векторов используйте размерность = 0. |
Возврат
- новый экземпляр ArgMax
public static ArgMax <TInt64> create (область действия , операнд <? расширяет TType > ввод, операнд <? расширяет TNumber > измерение)
Фабричный метод для создания класса, обертывающего новую операцию ArgMax, с использованием типов вывода по умолчанию.
Параметры
объем | текущий объем |
---|---|
измерение | int32 или int64 должен находиться в диапазоне `[-rank(input), Rank(input))`. Описывает, какое измерение входного тензора нужно уменьшить. Для векторов используйте размерность = 0. |
Возврат
- новый экземпляр ArgMax