[PyTorch] 파이토치 텐서의 최대값 및 그것의 인덱스

파이토치에서는 max 메소드를 이용해서 텐서(행렬)의 최대값을 구할 수 있습니다. 

 

 

만약에 행렬에서 각 열의 최대값들을 구하고 싶으면 X.max(dim=0)을, 각 행의 최대값을 구하고 싶으면 X.max(dim=1)을 입력해주시면 됩니다. 

 

 

이때는 최대값들 뿐만 아니라, 최대값의 인덱스들도 함께 반환됩니다.