Sigmoid 與 Softmax 函式
sigmoid 與 softmax 是深度學習中常見的啟動函式，通常用在神經網路的最後一步。
- Sigmoid 用於二元分類。
- Softmax 用於多類別分類。
給定網路的預啟動輸出張量，請套用合適的啟動函式以取得最終輸出。
torch.nn 已經以 nn 名稱匯入。
本練習屬於課程
使用 PyTorch 的深度學習入門
動手互動練習
試著完成這個範例程式碼，體驗一下這個練習。
input_tensor = torch.tensor([[2.4]])
# Create a sigmoid function and apply it on input_tensor
sigmoid = nn.____()
probability = ____(____)
print(probability)