PyTorch:Softmax 다 분류 실전 조작

다 중 분류 에서 자주 사용 하 는 방법 은 마지막 층 에 softmax 를 추가 하여 귀 일 화 하 는 것 이 고 값 이 가장 큰 차원 에 대응 하 는 위 치 는 이 견본 에 대응 하 는 유형 으로 한다.본 고 는 PyTorch 프레임 워 크 를 사용 하여 전형 적 인 이미지 데이터 세트 mnist 를 선택 하여 다 중 분 류 를 학습 합 니 다. softmax 함수 의 본질은 하나의 K ...

PyTorchSoftmax다 분류