tf.compat.v2.argmin
Возвращает индекс с наименьшим значением по осям тензора.
tf.compat.v2.argmin(
input, axis=None, output_type=tf.dtypes.int64, name=None
)
Обратите внимание, что в случае совпадений идентичность возвращаемого значения не гарантируется.
| Аргументы | |
|---|---|
input | A Tensor. Должен быть одного из следующих типов: float32, float64, int32, uint8, int16, int8, complex64, int64, qint8, quint8, qint32, bfloat16, uint16, complex128, half, uint32, uint64. |
axis | A Tensor. Должен быть одного из следующих типов: int32, int64. int32 или int64, должен быть в диапазоне -rank(input), rank(input)). Описывает ось входного тензора, по которой выполняется сокращение. Для векторов используйте ось = 0. |
output_type | Необязательный tf.DType из: tf.int32, tf.int64. По умолчанию tf.int64. |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
A Tensor типа output_type. |
Использование:
import tensorflow as tf a = [1, 10, 26.9, 2.8, 166.32, 62.3] b = tf.math.argmin(input = a) c = tf.keras.backend.eval(b) # c = 0 # here a[0] = 1 which is the smallest element of a across axis 0
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/compat/v2/argmin
