Fonction torch.argmin de PyTorch

Pytorch torch 参考手册 Manuel de référence Pytorch torch

torch.argminEst une fonction dans PyTorch utilisée pour retourner les indices des valeurs minimales le long d'une dimension.

Définition de fonction

torch.argmin(input, dim, keepdim)

Exemples d'utilisation

Exemple

import torch

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

print("Indice du minimum global :", torch.argmin(x))
print("Indice du minimum pour dim=1 :", torch.argmin(x, dim=1))

Le résultat de sortie est:

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

Pytorch torch 参考手册 Manuel de référence Pytorch torch

Autres extensions