① argmax는 몇 번 설명했습니다. max는 최댓값을 반환하고, argmax는 최댓값을 갖는 인덱스를 반환합니다. 다음 예제를 통해 최댓값과 인덱스에 대해 알아봅시다. 먼저 임의의 텐서를 생성합니다.
x = torch.rand(5,3) ------ (5×3) 형태의 텐서 생성 print(x)
다음과 같이 (5×3) 형태의 임의의 텐서가 생성되었습니다.
tensor([[0.6501, 0.6177, 0.5627], [0.3163, 0.5619, 0.6364], [0.8501, 0.9622, 0.4792], [0.1859, 0.6510, 0.1824], [0.4304, 0.1170, 0.8049]])
생성된 텐서를 이용하여 max와 argmax의 반환되는 값을 알아봅시다.
print(torch.max(x)) ------ max의 반환값 알아보기 print('---------------') print(torch.argmax(x)) ------ argmax의 반환값 알아보기
다음은 max와 argmax에 대한 반환값입니다. 다음 결과와 같이 max는 텐서의 요소 중 가장 높은 값을 반환하지만, argmax는 그 인덱스를 반환합니다.
tensor(0.9622) --------------- tensor(7)