① tensor.topk는 torch.argmax와 같은 효과입니다. 주어진 텐서에서 가장 큰 값의 인덱스(index)를 얻기 위해 사용합니다. 즉, 네트워크의 출력에서 가장 확률이 높은 값의 인덱스를 반환합니다. 다음 예시로 사용 방법을 확인할 수 있습니다.
import torch x = torch.arange(1., 6.) print(x) print('-----------------') print(torch.topk(x, 3)) ------ x 입력에서 가장 큰 값 세 개를 선택하여 그 값과 인덱스를 출력
출력 결과는 다음과 같습니다.
tensor([1., 2., 3., 4., 5.]) ----------------- torch.return_types.topk( values = tensor([5., 4., 3.]), indices = tensor([4, 3, 2]))
② t()는 차원 0과 1을 전치(transpose)하겠다는 의미입니다. 예를 들어 다음과 같이 사용됩니다.
x = torch.randn(3) ------ 1차원 텐서 생성 print(x) print(torch.t(x)) ------ 1차원 그대로 값을 반환 print('------------') x = torch.randn(2, 3) ------ (2, 3)차원 텐서 생성 print(x) print(torch.t(x)) ------ 입력이 전치되어 (3, 2)차원 반환