Función torch.argmin de PyTorch

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.argminEs una función en PyTorch para devolver el índice del valor mínimo a lo largo de una dimensión.

Definición de la función

torch.argmin(input, dim, keepdim)

Ejemplo de uso

Ejemplo

import torch

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

print("Índice mínimo global:", torch.argmin(x))
print("Índice mínimo para dim=1:", torch.argmin(x, dim=1))

El resultado de salida es:

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

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones