PyTorch torch.argmin Function
Pytorch torch Reference Manual
torch.argminIs a function in PyTorch used to return the indices of the minimum values along a dimension.
Function Definition
torch.argmin(input, dim, keepdim)
Usage Examples
Example
import torch
x = torch.tensor([[1, 3, 2], [4, 1, 3]])
print("Global minimum index:", torch.argmin(x))
print("dim=1 minimum index:", torch.argmin(x, dim=1))
x = torch.tensor([[1, 3, 2], [4, 1, 3]])
print("Global minimum index:", torch.argmin(x))
print("dim=1 minimum index:", torch.argmin(x, dim=1))
The output result is:
全局最小索引: tensor(0) dim=1 最小索引: tensor([0, 1])
Other Extensions