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