torch.sigmoid()、torch.nn.Sigmoid()和torch.nn.functional.sigmoid()三者之間的區別

在利用自定義損失函數進行損失計算的時候,需要使用到sigmoid函數。當在訓練模型的時候,使用torch.nn.Sigmoid()對輸出進行處理,但是提示__init__錯誤。我發現光sigmoid()就有三種調用的方式:torch.sigmoid(),torch.nn.Sigmoid()和torch.nn.functional.sigmoid()。 torch.sigmoid(): 這是一個方法
相關文章
相關標籤/搜索