神经网络的典型结构就是

__init__()

forward()

 

forward是自动调用的

        model = Net()
        y = model(x)
如上则调用网络模型定义的forward方法。

即,当把定义的网络模型model当作函数调用的时候就自动调用定义的网络模型的forward方法。

是通过nn.Module 的__call__方法调用的

 

就相当于调用了模型就是直接调用它的forward函数,y=model(x),这个x就是直接传入到forward函数的x参数

Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐