torch.max(input, dim, keepdim=False, ***, out=None)

torch.max(output,1)의 실행 결과는 output이 아래와 같다고 할 때,

output = torch.tensor([[2.5, 1.2, 0.3],
                       [0.1, 3.7, 0.2],
                       [0.2, 0.3, 2.1]])

각 행(dim=1)에서 최대값과 그 위치(인덱스)를 찾는다.

torch.return_types.max(
values=tensor([2.5000, 3.7000, 2.1000]),
indices=tensor([0, 1, 2]))