class Module(nn.Module):
def __init__(self):
super(Module, self).__init__()
def forward(self, x):
print("x====",x)
return x
def train_operation(self):
im_dat = 111
self(im_dat) # 在这里调用了forward方法
data = 122
# 实例化一个对象
module = Module()
# 前向传播
module.train_operation() # 方式一
# module(data) # 方式二
python两种方式调用forward
最新推荐文章于 2024-09-01 16:57:39 发布