인프런 커뮤니티 질문&답변
작성한 질문수
torch.armax에 관한 질문입니다
작성
·
552
·
수정됨
0
Multi-Class Classification 딥러닝 적용해보기 강의 8:40 부분에서 사용된
torch.argmax(x, dim=차원) 에 관한 질문입니다.
torch.argmax 함수의 자세한 실행 방식이 궁금해서 pytorch 공식 문서 등을 찾아보았는데요,
'dim' 파라미터에 관한 설명이 다음과 같이 나와 있었습니다.
dim (int) – the dimension to reduce. If
None, the argmax of the flattened input is returned.
여기서 "the dimension to reduce" 라는 표현이 정확히 무엇을 의미하는지 궁금합니다.
답변 1
0
잔재미코딩 DaveLee
지식공유자
안녕하세요. 답변 도우미입니다.
이야기하신 부분은 차원을 의미합니다. 영상에서 설명드렸듯이 차원이라는 용어가 혼용되긴 하는데요. 축 이라고 보시면 좋을 것 같아요. 다음 케이스를 작성해보시면 다음은 2차원인데, dim 이 0 일 때는 일종의 열, dim 이 1일 때는 일종의 행을 의미하고, 어떤 축을 기준으로 차원을 축소하겠냐는 의미입니다.
import torch
a = torch.FloatTensor([ [1, 2, 3], [4, 5, 6] ])
print (torch.argmax(a, dim=0))
print (torch.argmax(a, dim=1))
감사합니다.