인프런 커뮤니티 질문&답변
비전 트랜스포머 - 최신 모델' 강의에서 B*N*K 와 1*K*D 를 연산
작성
·
384
·
수정됨
답변 1
1
딥러닝호형
지식공유자
안녕하세요.
파이토치 nn.linear는 3차원 데이터에 대한 연산(배치 매트릭스 곱)을 할 수 있습니다. 각 배치 B개에 대한 N*K와 K*D 매트릭스 곱을 해주어 B*N*D라는 결과가 나오게 됩니다.
그리고 굵은R을 말씀하시는거죠? 실수 집합을 의미합니다. 즉, 해당 값이 크기가 ND이고 실수라는 의미입니다.
감사합니다.






