[일반] torch.nn.CrossEntropyLoss 질문있어요

[익명(180.233)]

label = torch.tensor([1,0])

predict = torch.tensor([1,0])

loss_fct = torch.nn.CrossEntropyLoss()

print(loss_fct(predict.float(), label.float()))

왜 0이 아닌 0.3133 값이 나오나요?

P(x)logQ(x) 이면 0+0 이어서 0이 나오지 않나요? 

[focalors(pytorch)]

프리딕트에 들어가는건 언노멀라이즈드 로짓이다~

[익명(223.62)]

감사합니다