tf.math.argmax

Возвращает индекс наибольшего значения по осям тензора.

Псевдонимы

Основные псевдонимы

tf.argmax

tf.math.argmax(
    input,
    axis=None,
    output_type=tf.dtypes.int64,
    name=None
)

Используется в блокнотах

Используется в руководстве	Используется в обучающих материалах
Многослойные перцептроны для распознавания цифр с помощью Core API Перенос метрик и оптимизаторов Распределённое обучение с помощью Core API и DTensor Введение в тензоры Введение в переменные	Трансферное обучение с YAMNet для классификации звуков окружающей среды Трансферное обучение с TensorFlow Hub Настраиваемое обучение: пошаговое руководство Распознавание звуков: распознавание ключевых слов Распределённое обучение с DTensor

Используется в руководстве

Используется в обучающих материалах

В случае равенства возвращается наименьший индекс.

Например:

A = tf.constant([2, 20, 30, 3, 6])
tf.math.argmax(A)  # A[2] is maximum in tensor A
<tf.Tensor: shape=(), dtype=int64, numpy=2>
B = tf.constant([[2, 20, 30, 3, 6], [3, 11, 16, 1, 8],
                 [14, 45, 23, 5, 27]])
tf.math.argmax(B, 0)
<tf.Tensor: shape=(5,), dtype=int64, numpy=array([2, 2, 0, 2, 2])>
tf.math.argmax(B, 1)
<tf.Tensor: shape=(3,), dtype=int64, numpy=array([2, 2, 1])>
C = tf.constant([0, 0, 0, 0])
tf.math.argmax(C) # Returns smallest index in case of ties
<tf.Tensor: shape=(), dtype=int64, numpy=0>

Аргументы
`input`	Объект типа `Tensor`.
`axis`	Целое число — ось, по которой выполняется свёртка. По умолчанию 0.
`output_type`	Необязательный тип данных результата (`tf.int32` или `tf.int64`). По умолчанию используется `tf.int64`.
`name`	Необязательное имя операции.

Возвращает
Объект `Tensor` типа `output_type`.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/math/argmax