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.
Bài tập này là một phần của khóa học
Nhập môn Deep Learning với PyTorch
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
input_tensor = torch.tensor([[2.4]])
# Create a sigmoid function and apply it on input_tensor
sigmoid = nn.____()
probability = ____(____)
print(probability)