Функция torch.amin в PyTorch

Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.aminЭто функция в PyTorch, используемая для возврата минимального значения тензора вдоль указанного измерения.

Определение функции

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

Пример использования

Пример

import torch

x = torch.tensor([[1, 3, 2], [4, 1, 3]])

# Вернуть минимальное значение всех элементов
print("Глобальный минимум:", torch.amin(x))

# Минимальное значение вдоль dim=0
print("минимум при dim=0:", torch.amin(x, dim=0))

# Минимальное значение вдоль dim=1
print("минимум при dim=1:", torch.amin(x, dim=1))

Результат вывода:

全局最小: tensor(1)
dim=0 最小: tensor([1, 1, 2])
dim=1 最小: tensor([1, 1])

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения