pytorch中继承nn.Module后类的奇怪用法

Question

我一直以为我看的代码有语法错误，直到我刚刚才发现他的玄机
先上代码：

import torch.nn as nn
class A():#nn.Module
    def __init__(self,a="MaXh",b=100):
        super(A, self).__init__()
        self.name = a
        self.age = b

    def forward(self,name2):
        print(f"Hello, I'm{self.name} and i am {self.age} yearsold","\n","my girlfriend is",name2)
        return name2


class B:
    def __init__(self,c=0):
        self.a = A()

    def forward(self,name2):
        out=self.a(name2)#看似有语法错误，不应该是out=self.a.forward(name2) ?
        return out

b = B()
b.forward("ma")

运行结果：报错
但是，如果让A类继承nn.Module:

import torch.nn as nn
class A(nn.Module):#
    def __init__(self,a="MaXh",b=100):
        super(A, self).__init__()
        self.name = a
        self.age = b

    def forward(self,name2):
        print(f"Hello, I'm{self.name} and i am {self.age} yearsold","\n","my girlfriend is",name2)
        return name2


class B:
    def __init__(self,c=0):
        self.a = A()

    def forward(self,name2):
        out=self.a(name2)#看似有语法错误，不应该是out=self.a.forward(name2) ?
        return out

b = B()
b.forward("ma")

代码中看似有语法错误的那一行就能正常运行：

Hello, I'mMaXh and i am 100 yearsold 
 my girlfriend is ma
'ma'

Why?

Answer 1

基于Monster 组和GPT的调写：

• 在第一个代码示例中，A类没有继承自nn.Module，因此它不是一个PyTorch的模块，而是一个普通的Python类。在这种情况下，forward方法只是一个普通的实例方法，并不具有特殊的含义。

• 当你调用self.a(name2)时，self.a是一个A类的实例，而不是一个PyTorch模块，所以它不会自动调用forward方法。因此，你会得到一个TypeError，因为你不能像调用函数一样调用一个非函数的对象。

• 在第二个代码示例中，A类继承自nn.Module，它成为了一个PyTorch的模块。PyTorch模块中的forward方法具有特殊的含义，用于定义模块的前向传播逻辑。

• 因此，在第二个示例中，当你调用self.a(name2)时，由于self.a是一个PyTorch模块，它会自动调用其forward方法，从而正常运行代码并输出结果。