让python类直接被调用

让python类直接被调用

之前在pytorch和keras中经常发现一个类model被直接调用，发现很有意思。于是就去看了看pytorch中nn.Module的源码，发现是定义了__call__(self)函数再去调用forward()函数。举个例子如下:

import math
class Pow(object):
def __init__(self,n=2):
self.n=n
super(Pow,self).__init__()
def forward(self,x):
return math.pow(x,self.n)
def __call__(self,x):
return self.forward(x)
l=Pow(2)
y=l(10)
print(y) #输出结果是100
l=Pow(3)
y=l(10)
print(y) #输出结果是1000