Pytorch nn.Module和forward的参数关系

module

在定义好module类之后，调用module时，通常执行
一个参数x

pred = module(x)

但也有执行
两个参数的情况

pred = module(x,y)

forward

import torch
class Net(torch.nn.Module):
    def __init__(self):
        super(Net, self).__init__()
        print("init执行")

    def forward(self,x,out):
        x=out
        print("输入为：",x)
        print("out:",out)
        return x

module = Net()
pred = module(5,6)
print(pred)

输出为：

init执行
输入为： 6
out: 6
6

• 调用module时传入参数与module类定义的forward参数一致
• module返回的函数值为forward的返回值，即return的x=out，即out值，这里返回了6，于是pred为6