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

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

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

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

torch.argmin(input, dim, keepdim)

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

Пример

import torch

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

print("Индекс глобального минимума:", torch.argmin(x))
print("индекс минимума при dim=1:", torch.argmin(x, dim=1))

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

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

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

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