데이터분석/Pytorch7 [Pytorch] 파이토치 Tensor(텐서)의 최대 값, 인덱스 sample_tensor.max(dim=0) # dim=0은 행에서 최대값, 인덱스 sample_tensor.max(dim=1) # dim=1은 열에서 최대값, 인덱스 2022. 11. 28. 이전 1 2 다음 728x90
