Функция min
Функция min возвращает минимальное значение среди всех элементов тензора.
Если указан параметр dim, то функция возвращает минимальные значения вдоль указанной размерности,
а также их индексы. Для тензоров с комплексными числами используется модуль числа.
Синтаксис
torch.min(input)
torch.min(input, dim, keepdim=False)
Параметры:
-
input- входной тензор. -
dim- размерность, вдоль которой ищется минимум. -
keepdim- еслиTrue, то выходной тензор будет иметь ту же размерность, что и входной (уменьшенная размерность будет равна 1).
Пример
Давайте найдём минимальное значение в одномерном тензоре:
import torch
t = torch.tensor([5, 2, 8, 1, 9])
res = torch.min(t)
print(res)
Результат выполнения кода:
tensor(1)
Пример
Найдём минимальное значение в двумерном тензоре:
import torch
t = torch.tensor([
[3, 7, 1],
[9, 2, 5],
])
res = torch.min(t)
print(res)
Результат выполнения кода:
tensor(1)
Пример
Найдём минимальные значения вдоль строк (по столбцам) с помощью параметра dim:
import torch
t = torch.tensor([
[3, 7, 1],
[9, 2, 5],
])
res, indices = torch.min(t, dim=1)
print(res)
print(indices)
Результат выполнения кода:
tensor([1, 2])
tensor([2, 1])
В результате получаем минимальные значения для каждой строки и их индексы.
Пример
Используем параметр keepdim для сохранения размерности:
import torch
t = torch.tensor([
[3, 7, 1],
[9, 2, 5],
])
res, indices = torch.min(t, dim=1, keepdim=True)
print(res)
print(res.shape)
Результат выполнения кода:
tensor([[1],
[2]])
torch.Size([2, 1])
