torch의 CrossEntropyLoss

torch.nn.CrossEntropyLoss

softmax + CEloss

앞단의 노드를 클래스 개수만큼 구성 -> loss func에 바로 넣어주면된다.

torch.nn.BCELoss

only CEloss
이진분류시 사용.
softmax 함수가 포함되지 않은 형태

앞단에서 한개의 노드 -> sigmoid -> loss func 사용

BCEWithLogitsLoss

sigmoid + CEloss

앞단에서 한개의 노드 -> loss func 사용