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

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

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

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

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

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

Пример

import torch

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

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

# Максимум по dim=0
print("максимум по dim=0:", torch.amax(x, dim=0))

# Максимум по dim=1
print("максимум по dim=1:", torch.amax(x, dim=1))

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

全局最大: tensor(4)
dim=0 最大: tensor([4, 3, 3])
dim=1 最大: tensor([3, 4])

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

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