pytorch 中的 forward 的使用与解释

转自：https://blog.csdn.net/xu380393916/article/details/97280035，感谢作者分享。 在使用pytorch的时候，模型训练时，不需要使用forward，只要在实例化一个对象中传入对应的参数就可以自动调用 forward 函数 class Module(nn.Module):def __init__(self):super(Module, self).__init__()# 0 Comments

转自：https://blog.csdn.net/xu380393916/article/details/97280035，感谢作者分享。
在使用pytorch的时候，模型训练时，不需要使用forward，只要在实例化一个对象中传入对应的参数就可以自动调用 forward 函数

class Module(nn.Module):def __init__(self):super(Module, self).__init__()# ......def forward(self, x):# ......return xdata = .....  #输入数据
# 实例化一个对象
module = Module()
# 前向传播
module(data)  
# 而不是使用下面的
# module.forward(data)

实际上

module(data)

是等价于

module.forward(data)

本文链接：https://my.lmcjl.com/post/1922.html

展开阅读全文

相关文章

4 评论

留下您的评论.