(1)设置文字行数为2行(多行显礻)
(1)单行显示超出为省略号.
(1)设置文字行数为2行(多行显礻)
(1)单行显示超出为省略号.
下面是利用PyTorch定义深度网络层(Op)示例:
由上例代码可以看到不论是在定义网络结构还是定义网络层的操作(Op),均需要定义forward函数下面看一下对PyTorch的forward方法的描述:
那么调用forward方法的具体流程是什么样的呢?
上述中“调用module的call方法”是指nn.Module 的__call__方法定义__call__方法嘚类可以当作函数调用,具体参考Python的面向对象编程也就是说,当把定义的网络模型model当作函数调用的时候就自动调用定义的网络模型的forward方法nn.Module 的__call__方法部分源码如下所示:
可以看到,当执行model(x)的时候底层自动调用forward方法计算结果。具体示例如下:
如上则调用网络模型定义的forward方法
如果您觉得我的文章对您有所帮助,欢迎扫码进行赞赏!