Функция amin

Функция amin возвращает минимальное значение тензора. Если не указать размерность, то она вернет минимальный элемент среди всех элементов. Если указать параметр dim, то поиск будет выполняться вдоль заданной размерности. Результат может быть как скалярным значением, так и тензором с уменьшенной размерностью.

Синтаксис

torch.amin(input, dim=None, keepdim=False)

Параметры функции:

input (Tensor) - исходный тензор;
dim (int или tuple, опционально) - размерность, вдоль которой ищем минимум. Если не указан, ищем глобальный минимум;
keepdim (bool, опционально, по умолчанию False) - сохранять ли размерность.

Пример

Найдем минимальное значение среди всех элементов одномерного тензора:

import torch

t = torch.tensor([5, 2, 8, 1, 3])
res = torch.amin(t)
print(res)

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

tensor(1)

Пример

Найдем минимальное значение для двумерного тензора:

import torch

t = torch.tensor([
	[4, 9, 2],
	[7, 3, 6],
])
res = torch.amin(t)
print(res)

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

tensor(2)

Пример

Найдем минимальные значения вдоль строк (по столбцам), указав размерность dim=0:

import torch

t = torch.tensor([
	[4, 9, 2],
	[7, 3, 6],
])
res = torch.amin(t, dim=0)
print(res)

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

tensor([4, 3, 2])

Пример

Найдем минимальные значения вдоль столбцов (по строкам), указав размерность dim=1:

import torch

t = torch.tensor([
	[4, 9, 2],
	[7, 3, 6],
])
res = torch.amin(t, dim=1)
print(res)

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

tensor([2, 3])

Пример

Используем параметр keepdim, чтобы сохранить размерность тензора:

import torch

t = torch.tensor([
	[4, 9, 2],
	[7, 3, 6],
])
res = torch.amin(t, dim=1, keepdim=True)
print(res)

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

tensor([
	[2],
	[3],
])

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

функцию amax,
которая находит максимальное значение тензора
функцию min,
которая также находит минимум, но может принимать два тензора для поэлементного сравнения
функцию argmin,
которая возвращает индекс минимального элемента
функцию clamp,
которая ограничивает значения тензора заданным диапазоном