[Pytorch] 파이토치 Tensor(텐서)의 최대 값, 인덱스

 

sample_tensor.max(dim=0) # dim=0은 행에서 최대값, 인덱스
sample_tensor.max(dim=1) # dim=1은 열에서 최대값, 인덱스