torch.max

텐서 에 있는 모든 요소의 최대값 반환
demension이 주어질 경우, (values, indices) tuple 반환

사용법

dim : 1 (→ 방향) , 0 (↓ 방향)

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