Функция PyTorch torch.aminmax

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

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

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

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

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

Пример

import torch

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

# Возвращает минимальное и максимальное значения всех элементов
mm = torch.aminmax(x)
print("Минимальное значение:", mm[0])
print("Максимальное значение:", mm[1])

# Минимальное и максимальное значения вдоль dim=0
mm0 = torch.aminmax(x, dim=0)
print("минимум при dim=0:", mm0[0])
print("максимум при dim=0:", mm0[1])

# Минимальное и максимальное значения вдоль dim=1
mm1 = torch.aminmax(x, dim=1)
print("минимум при dim=1:", mm1[0])
print("максимум при dim=1:", mm1[1])

Вывод:

最小值: tensor(1)
最大值: tensor(4)
dim=0 最小值: tensor([1, 1, 2])
dim=0 最大值: tensor([4, 3, 3])
dim=1 最小值: tensor([1, 1])
dim=1 最大值: tensor([3, 4])

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

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