1. 학습
    2. /
  2. 강의
    3. /
  3. PyTorch로 배우는 딥러닝 입문
Connected

연습 문제

시그모이드와 소프트맥스 함수

딥러닝에서 시그모이드와 소프트맥스 함수는 중요한 활성화 함수로, 신경망의 마지막 단계에서 자주 사용돼요.

  • Sigmoid는 이진 분류에 사용돼요
  • Softmax는 다중 클래스 분류에 사용돼요

네트워크에서 나온 사전 활성화 출력 텐서가 주어졌을 때, 적절한 활성화 함수를 적용해 최종 출력을 얻으세요.

torch.nn은 이미 nn으로 임포트되어 있어요.

지침 1/2

undefined XP
  • 1
    • 시그모이드 함수를 만들고 input_tensor에 적용하여 이진 분류 작업의 확률을 생성하세요.
  • 2
    • 소프트맥스 함수를 만들고 input_tensor에 적용하여 다중 클래스 분류 작업의 확률을 생성하세요.