Функция argmin

Функция argmin возвращает индекс минимального значения вдоль заданной оси тензора. Первым параметром функция принимает тензор, вторым параметром можно передать ось, по которой выполняется поиск. Если ось не указана, поиск выполняется по всем элементам тензора.

Синтаксис

tf.argmin(input, axis=None, output_type=tf.int64)

Пример

Давайте найдем индекс минимального элемента в одномерном тензоре:

import tensorflow as tf

t = tf.constant([3, 1, 5, 2, 4])
res = tf.argmin(t)
print(res)

Результат выполнения кода:

tf.Tensor([1], shape=(1,), dtype=int64)

Пример

Давайте найдем индексы минимальных элементов по столбцам двумерного тензора:

import tensorflow as tf

t = tf.constant([[1, 2, 3], [4, 5, 6]])
res = tf.argmin(t, axis=0)
print(res)

Результат выполнения кода:

tf.Tensor([0 0 0], shape=(3,), dtype=int64)

Пример

Давайте найдем индексы минимальных элементов по строкам двумерного тензора:

import tensorflow as tf

t = tf.constant([[1, 2, 3], [4, 5, 6]])
res = tf.argmin(t, axis=1)
print(res)

Результат выполнения кода:

tf.Tensor([0 0], shape=(2,), dtype=int64)

Пример

Давайте получим результат в виде тензора типа int32:

import tensorflow as tf

t = tf.constant([3, 1, 5, 2, 4])
res = tf.argmin(t, output_type=tf.int32)
print(res)

Результат выполнения кода:

tf.Tensor([1], shape=(1,), dtype=int32)

Пример

Давайте извлечем индекс из результата и преобразуем его в обычное число:

import tensorflow as tf

t = tf.constant([3, 1, 5, 2, 4])
res = tf.argmin(t).numpy()[0]
print(res)

Результат выполнения кода:

1

Смотрите также

функцию argmax,
которая возвращает индекс максимального элемента
функцию reduce_min,
которая возвращает минимальное значение тензора
функцию sort,
которая сортирует элементы тензора
функцию argsort,
которая возвращает индексы отсортированных элементов