torch.armax에 관한 질문입니다
654
45 asked
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" 라는 표현이 정확히 무엇을 의미하는지 궁금합니다.
Answer 1
0
안녕하세요. 답변 도우미입니다.
이야기하신 부분은 차원을 의미합니다. 영상에서 설명드렸듯이 차원이라는 용어가 혼용되긴 하는데요. 축 이라고 보시면 좋을 것 같아요. 다음 케이스를 작성해보시면 다음은 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))
감사합니다.
import torch가 안되는 경우는 어떻게 하나요?
0
16
1
강의 필기는 어떻게 하나요?
0
84
1
섹션 5퀴즈 4번 문제 보기 오류로 보임
0
62
1
섹션17 81번이랑 82번 강의가 중복되는 것 같아서 질문드립니다.
0
88
2
강의교안이 안 옵니다
0
80
1
17-2강 Transfer learning 실습 관련 질문
0
134
2
13섹션 강의자료가 없네요^^
0
92
1
강의자료가 없네요.
0
105
2
LSTM 모델 학습 관련한 질문입니다.
0
127
1
MSE LOSS 관련
0
131
1
test시 minibatch 사용?
0
133
1
Average Test loss 계산식 문의
0
101
1
파이토치 설치
0
247
1
[Pytorch 기울기의 누적 곱? 누적 합?]
0
135
1
3d 텐서에서의 축 구분 질문
0
119
1
항상 tensor([0., 0., 0.]) 형식으로만 나오는 이유
0
111
1
강의자료 PDF 다운로드에 관하여
0
132
1
[실무에서 판다스 copy()메서드의 깊은복사 얕은복사 조정 소요가 생길까?]
0
116
1
섹션 다양한 Optimizer 정리하기 중 딥러닝으로 Regression 문제 적용해보기 (House Price Kaggle 문제) 강의에서 오류가 발생합니다.
0
148
1
마지막에 confusion matrix 작성할때
0
157
1
아나콘다를 사용하지 않고 파이토치 사용하는 방법
0
853
1
test data 의 loss 계산식 문의
0
288
1
차원 출력 관련 문의
0
245
1
섹션 7-4 당뇨병
0
387
3