PyTorch(1) torch.nn與torch.nn.functional之間的區別和聯繫

在閱讀PyTorch官網的教程的時候,發現介紹如何利用Pytorch搭建一個神經網絡的示例代碼是這樣寫。 迷惑的地方是在於forward的函數的定義方法。爲什麼要把網絡中的一部分層在__init__()函數裏定義出來,而另一部分層則是在__forward()__函數裏定義?並且一個用的是nn,另一個用的是nn.functional。同一種層的API定義有兩種,這樣看似冗餘的設計是爲了什麼呢? 通過
相關文章
相關標籤/搜索