실행하면 다음과 같이 출력됩니다.
11
또한, torch.max(x[2], dim=0) 의미는 다음과 같습니다.
x = torch.randn([4, 4]) ------ (4×4) 크기를 갖는 임의의 텐서 생성 print(x) max_elements, max_idxs = torch.max(x, dim=0) ------ torch.max 값을 가지고 오되 dim=0(행을 기준)으로 최댓값을 가져옵니다. print(max_elements) print(max_idxs)
코드를 실행한 결과 행을 기준(dim=0)으로 최댓값을 가져온 것을 확인할 수 있습니다.
tensor([[-0.3027, -0.3185, 0.4117, 0.4915], [-3.1112, 1.3952, -0.2522, -0.5341], [-0.5381, -0.2026, -1.0494, -1.2712], [-0.3689, -1.8523, 2.0229, 0.5725]]) tensor([-0.3027, 1.3952, 2.0229, 0.5725]) tensor([0, 1, 3, 3])
다음은 본래 이미지를 출력하기 위한 함수입니다. 이미 앞에서 이미지의 크기 및 전처리 과정 등을 거쳤기 때문에 이 상태에서 이미지를 출력하면 다음 그림과 같습니다.
▲ 그림 6-20 본래 이미지 색상이 왜곡됨