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

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

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

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

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

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

Пример

import torch

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

print(Глобальный минимум:, torch.min(x))
print(минимум при dim=0:, torch.min(x, dim=0))

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

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