인프런 커뮤니티 질문&답변
cross_entropy
작성
·
2.4K
답변 1
0
Justin
지식공유자
안녕하세요, Justin입니다.
cross_entropy 공식을 계산할 때 입력값의 shape이 맞지 않아서 발생되는 문제로 판단됩니다.
cross_entropy의 입력값인 prediction, target의 shape을 확인 후 1D tensor로 맞추어 실행하셔야 될 것 같습니다.
감사합니다.
작성
·
2.4K
답변 1
0
안녕하세요, Justin입니다.
cross_entropy 공식을 계산할 때 입력값의 shape이 맞지 않아서 발생되는 문제로 판단됩니다.
cross_entropy의 입력값인 prediction, target의 shape을 확인 후 1D tensor로 맞추어 실행하셔야 될 것 같습니다.
감사합니다.