開始使用免費開始

實作 ReLU

整流線性單元(ReLU)是深度學習中廣泛使用的啟用函式,可用來解決像是梯度消失等問題。

在這個練習中,你將在 PyTorch 中實作 ReLU,將它套用到「正值與負值」上,並觀察結果。

torch.nn 套件已替你匯入為 nn

本練習屬於課程

使用 PyTorch 的深度學習入門

檢視課程

動手互動練習

試著完成這個範例程式碼,體驗一下這個練習。

# Create a ReLU function with PyTorch
relu_pytorch = ____
編輯並執行程式碼