在Python中，`__call__`是一个特殊的方法，它允许将一个类的实例像函数一样调用，使对象具有函数的行为，在某些情况下非常有用[^1]。 ### 原理 当创建一个类的实例后，如果该类实现了`__call__`方法，那么这个实例就可以像函数一样被调用，调用实例时实际上就是调用了`__call__`方法。 ### 使用示例 #### 简单示例 ```python class MyClass: def __call__(self, x, y): return x + y # 创建实例 obj = MyClass() # 像函数一样调用实例 result = obj(3, 5) print(result) # 输出: 8 ``` 在这个示例中，`MyClass`类实现了`__call__`方法，它接受两个参数并返回它们的和。创建`MyClass`的实例`obj`后，可以像调用函数一样调用`obj`，实际上就是调用了`__call__`方法。 #### 在PyTorch中的应用 在PyTorch中，自定义的类继承于`nn.Module`父类，且子类中对父类的`forward`方法进行了重载，父类`nn.Module`中的`__call__`调用了`forward`方法，所以可以直接用子类的实例作为可调用对象[^2]。示例代码如下： ```python import torch import torch.nn as nn class MyModel(nn.Module): def __init__(self): super(MyModel, self).__init__() self.fc = nn.Linear(10, 1) def forward(self, x): return self.fc(x) # 创建模型实例 model = MyModel() # 创建输入数据 input_tensor = torch.randn(1, 10) # 像函数一样调用模型实例 output = model(input_tensor) print(output) ``` 在这个示例中，`MyModel`类继承自`nn.Module`，并重载了`forward`方法。创建`MyModel`的实例`model`后，可以像调用函数一样调用`model`，实际上是调用了`__call__`方法，而`__call__`方法又调用了`forward`方法。 #### 类装饰器示例 如果需要将一个类作为装饰器，需要为这个类实现`__call__`方法。示例代码如下： ```python class TmpTest: def __init__(self, func): self.func = func def __call__(self, *args, **kwargs): print("Before function call") result = self.func(*args, **kwargs) print("After function call") return result @TmpTest def my_function(): print("Inside function") my_function() ``` 在这个示例中，`TmpTest`类作为装饰器，实现了`__call__`方法。`my_function`被`TmpTest`装饰后，调用`my_function`时实际上是调用了`TmpTest`实例的`__call__`方法。