tf.keras.ops.argmin

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

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

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

tf.keras.ops.numpy.argmin

tf.keras.ops.argmin(
    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.argmin(x)
array(0, dtype=int32)
keras.ops.argmin(x, axis=0)
array([0, 0, 0], dtype=int32)
keras.ops.argmin(x, axis=1)
array([0, 0], 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/argmin