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

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

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

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

torch.max(input, dim, keepdim, out)

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

Пример

import torch

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

# Глобальный максимум
print("Глобальный максимум:", torch.max(x))

# Максимум по измерению
print("Максимум при dim=0:", torch.max(x, dim=0))

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

全局最大: tensor(6)
dim=0 最大: torch.return_types.max(values=tensor([4, 5, 6]), indices=tensor([1, 1, 1]))

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

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