tf.keras.ops.argmax

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

Просмотреть псевдонимы

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

tf.keras.ops.numpy.argmax

tf.keras.ops.argmax(
    x, axis=None, keepdims=False
)

Аргументы
`x`	Входной тензор.
`axis`	По умолчанию индекс относится к сплющенному тензору, иначе — к указанной оси.
`keepdims`	Если задано значение `True`, оси, по которым выполняется сведение, остаются в результате в качестве измерений с размером один. Значение по умолчанию — `False`.

Возвращает
Тензор индексов. Он имеет ту же форму, что и `x`, но без измерения вдоль `axis`.

Пример:

x = keras.ops.arange(6).reshape(2, 3) + 10
x
array([[10, 11, 12],
       [13, 14, 15]], dtype=int32)
keras.ops.argmax(x)
array(5, dtype=int32)
keras.ops.argmax(x, axis=0)
array([1, 1, 1], dtype=int32)
keras.ops.argmax(x, axis=1)
array([2, 2], dtype=int32)

© 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/keras/ops/argmax