YouTipPytorch Torch Amax
# PyTorch torch.amax Function
* * Pytorch torch reference manual](#)
`torch.amax` is a function in PyTorch used to return the maximum value of a tensor along a specified dimension.
### Function Definition
torch.amax(input, dim, keepdim=False)
* * *
## Usage Example
## Example
import torch
x = torch.tensor([[1,3,2],[4,1,3]])
# Return the maximum value of all elements
print("Global maximum:", torch.amax(x))
# Maximum along dim=0
print("dim=0 maximum:", torch.amax(x, dim=0))
# Maximum along dim=1
print("dim=1 maximum:", torch.amax(x, dim=1))
Output:
Global maximum: tensor(4)
dim=0 maximum: tensor([4, 3, 3])
dim=1 maximum: tensor([3, 4])
* * Pytorch torch reference manual](#)