如何在 python 中调用没有方法名称的方法

Question

class LinearRegression(nn.Module):
def __init__(self,input_size,output_size):
    # super function. It inherits from nn.Module and we can access everythink in nn.Module
    super(LinearRegression,self).__init__()
    # Linear function.
    self.linear = nn.Linear(input_dim,output_dim)

def forward(self,x):
    return self.linear(x)


input_dim = 1
output_dim = 1
model = LinearRegression(input_dim,output_dim)

在这个代码块中，当我想在 model object 中调用 forward 方法时，我可以调用 2 种不同的方式

第一种方式

results = model.forward(car_price_tensor)

第二种方式

results = model(car_price_tensor)

我在不同的 class 和 object 上尝试了第二种方法，但它不起作用。 怎么可能？

Answer 1

这种行为允许您调用 Python object 就像 function 一样，是通过特殊方法__call__启用的，这里有一些解释。

torch.nn.Module是实现此行为的 class。 您LinearRegression是它的子类，因此它继承了此行为。 默认情况下， __call__方法是从您的forward实现派生的，但并不相同。

这里解释了行为上的差异。