如何在 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實現派生的，但並不相同。

這里解釋了行為上的差異。