标签：__ 函数 nn 子类 Module call module forward
使用
直接通过类的实例对象就可以向类中的forward函数进行参数的传递(当然也可以通过调用forward函数进行传参)
import torch.nn as nn
class MyModule(nn.Module):
def __init__(self):
super().__init__()
def forward(self, x):
return x
data1 = 1
data2 = 2
module = MyModule()
x1 = module(data1) # 不需要显示调用forward函数就可以传递参数
x2 = module.forward(data2)
print(x1)
print(x2)
>> 1
>> 2
解释
nn.Module() 中包含了 __call__ 函数；
实现了 __call__ 函数的类，其类实例是一个可调用的对象，其可以简化对于类中某些方法的调用（写在__call__ 中的方法），模糊了实例对象和类成员函数的区别。使用类实例 module() 时 就相当于 module.__call__()，如果在 __call()__ 中写上函数，就可以直接通过类实例对象传参调用了。
而在 nn.Module() 中的 __call__ 函数中调用了 forward() 函数，
...
# 例子 #
def __call__(self, param):
res = self.forward(param)
return res
...
由于继承关系，对于MyModule(nn.Module) 类 同样具备了 __call__ 函数的功能，即可以通过类实例module 直接 调用 forward 并传参。
标签：__,函数,nn,子类,Module,call,module,forward 来源： https://www.cnblogs.com/jack-nie-23/p/16506630.html
本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享； 2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关； 3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关； 4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除； 5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。