pytorch의 CrossEntropyLoss 를 구현한 custom Loss에 관하여 질문드립니다.

[Sung_Sue_Hwang]

코드가 모두 나와 있는 것은 아니라 제가 예시로 한번 해봤는데 문제는 없어 보입니다.

결과가 동일하게 나오는데요 혹시 다른 부분의 문제가 있는 것은 아닐까요?