1.3 nn.Module 神經網絡(一)

torch.nn是專門爲神經網絡設計的模塊化接口。nn構建於Autograd之上,用來定義和運行神經網絡。nn.Module是nn中最重要的類,可以把它看作一個網絡的封裝,包含網絡各層定義以及forward方法、調用forward(input)方法,可返回前向傳播結果。下面以LeNet網絡爲例:                                                    
相關文章
相關標籤/搜索