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