YouTipPytorch Torch Mode
# PyTorch torch.mode Function
* * *
[
* * *
## Usage Examples
## Example
import torch
x = torch.tensor([1,2,2,3,3,3])
# Return mode
print("Mode:", torch.mode(x))
# Mode along dim=0
y = torch.tensor([[1,2,2],[1,2,3]])
print("dim=0 Mode:", torch.mode(y, dim=0))
# Mode along dim=1
print("dim=1 Mode:", torch.mode(y, dim=1))
Output result:
Mode: torch.return_types.mode(values=tensor(3), indices=tensor(4)) dim=0 Mode: torch.return_types.mode(values=tensor([1, 2, 2]), indices=tensor([0, 0, 0])) dim=1 Mode: torch.return_types.mode(values=tensor([2, 3]), indices=tensor([1, 2]))
* * *
[![Image 4: Pytorch torch Reference Manual]( Pytorch torch Reference Manual](