PyTorch torch.amin Function
Pytorch torch Reference Manual
torch.aminIt is a function in PyTorch used to return the minimum value of a tensor along a specified dimension.
Function Definition
torch.amin(input, dim, keepdim=False)
Usage Example
Example
import torch
x = torch.tensor([[1, 3, 2], [4, 1, 3]])
# Return the minimum value of all elements
print("Global minimum:", torch.amin(x))
# Minimum value along dim=0
print("dim=0 minimum:", torch.amin(x, dim=0))
# Minimum value along dim=1
print("dim=1 minimum:", torch.amin(x, dim=1))
x = torch.tensor([[1, 3, 2], [4, 1, 3]])
# Return the minimum value of all elements
print("Global minimum:", torch.amin(x))
# Minimum value along dim=0
print("dim=0 minimum:", torch.amin(x, dim=0))
# Minimum value along dim=1
print("dim=1 minimum:", torch.amin(x, dim=1))
The output result is:
全局最小: tensor(1) dim=0 最小: tensor([1, 1, 2]) dim=1 最小: tensor([1, 1])
Other Extensions