PyTorch深度学习实践05用pytorch实现线性回归 __call__ 和forward函数
PyTorch深度学习实践05用pytorch实现线性回归 call 和forward函数
__call __
先上结论:
model(x_data)到底是怎么调用的:
相当于model(x_data)实际上调用的是__call__(x_data),这里的call函数是LinearModel继承自torch.nn.Module的call。也就是Module这个基类的call函数。然后__call__(input)调用(Module的)forward函数。但是我们需要的forward函数和基类里的forwar
原创
2022-02-24 16:36:57 ·
828 阅读 ·
1 评论