인프런 영문 브랜드 로고
인프런 영문 브랜드 로고

인프런 커뮤니티 질문&답변

byeongal님의 프로필 이미지

작성한 질문수

[PyTorch] 쉽고 빠르게 배우는 딥러닝

[실습] RNN을 이용한 영화 리뷰 예측 모델 만들기

cross_entropy

작성

·

2.3K

0

RNN의 cross_entropy를 적용해서 실행시켰는데 1D target tensor expected, multi-target not supported 라고 나옵니다. 코드는 모두 맞게 썻습니다. 이유를 알고싶습니다.

답변 1

0

Justin님의 프로필 이미지
Justin
지식공유자

안녕하세요, Justin입니다.

cross_entropy 공식을 계산할 때 입력값의 shape이 맞지 않아서 발생되는 문제로 판단됩니다.
cross_entropy의 입력값인 prediction, target의 shape을 확인 후 1D tensor로 맞추어 실행하셔야 될 것 같습니다.

감사합니다.

byeongal님의 프로필 이미지

작성한 질문수

질문하기