1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Nhập môn Deep Learning với PyTorch
Connected

Bài tập

Hàm sigmoid và softmax

Hàm sigmoid và softmax là các hàm kích hoạt quan trọng trong deep learning, thường dùng ở bước cuối của mạng nơ-ron.

  • Sigmoid dùng cho bài toán phân loại nhị phân
  • Softmax dùng cho bài toán phân loại đa lớp

Với một tensor đầu ra trước kích hoạt từ mạng, hãy áp dụng hàm kích hoạt phù hợp để thu được đầu ra cuối cùng.

torch.nn đã được import dưới tên nn.

Hướng dẫn 1/2

undefined XP
  • 1
    • Tạo một hàm sigmoid và áp dụng lên input_tensor để tạo xác suất cho tác vụ phân loại nhị phân.
  • 2
    • Tạo một hàm softmax và áp dụng lên input_tensor để tạo xác suất cho tác vụ phân loại đa lớp.