在使用pytorch框架时,难免要自己定义网络。于是,super(XXXX, self).init(),就成了自定义网络结构时必不可少的第一句。但是,super(XXXX, self).init()具体的作用是什么我一直没有搞清楚。阅读了大量的博客后,我终于搞懂了!
一言以蔽之:super(XXX, self).init()——对继承自父类的属性进行初始化,并且用父类的初始化方法初始化继承的属性。
我们先看一个简单的例子:

class Person():    def __init__(self, name, gender):    # 为name和gender赋值         self.name = name        self.gender = gender            def printinfo(self):            print(self.name, self.gender)# Stu类继承Person类        class Stu(Person):    def __init__(self, name, gender, school):            # 使用父类的初始化方法来初始化子类name和gender属性        super(Stu, self).__init__(name, gender)                  self.school = school            def printinfo(self):       # 对父类的printinfo方法进行重写               print(self.name, self.gender, self.school)     if __name__=='__main__':    stu = Stu('Bob', 'female', '5th')    stu.printinfo()>>>Bob female 5th

当然,如果初始化的逻辑与父类的不同,不使用父类的方法,自己重新初始化也是可以的。比如:

class Person(object):    def __init__(self, name, gender, age):        self.name = name        self.gender = gender        self.age = age        # Student类继承Person类class Student(Person):    def __init__(self, name, gender, age, school, score):        # 调用父类的初始化方法,初始化name、gender、age属性        super(Student, self).__init__(name,gender, age)                # 对name、gender属性进行改写。age属性仍保持父类的初始化方法                self.name = name.upper()  # 姓名改为大写        self.gender = gender.upper()  # 性别改为大写        self.school = school        self.score = score        s = Student("Alice", "female", "18", "High school", "17")print(s.name, s.gender, s.school, s.score)>>>ALICE FEMALE High school 17

在理解了上面的小例子后,我们再仔细研究一下机器学习中的代码:

class Net(nn.Module): # 继承自nn.Moudle    def __init__(self):            super(Net, self).__init__()        # 输入图像channel:1;输出channel:6;5x5卷积核        self.conv1 = nn.Conv2d(1, 6, 5)

super(Net, self).init()的含义:子类Net类继承父类nn.Module,super(Net, self).init()就是对继承自父类nn.Module的属性进行初始化。并且是用nn.Module的初始化方法来初始化继承的属性。

也就是:用父类的方法初始化子类的属性。

有的人肯定会问,为啥要用父类的方法去初始化属性呢?原因很简单:就是因为父类的方法已经写好了,我们只需要调用就可以了。难道你还想自己写一堆代码去初始化各种权重和参数,处理一堆forward和backward的逻辑吗?

最后,多一句嘴,介绍一写在python中__init()的作用
在python中创建类后,通常会创建一个
init__ ()方法,这个方法会在创建类的实例的时候自动执行

实例1:【实例化Bob这个对象的时候, __ init__ ()方法会自动执行】:
在下面的示例中,我们在实例化Bob这个对象的时候, __ init__ ()方法就已经自动执行了,如果不是__ init__ ()方法,比如说eat()方法,那就只有在调用时才会执行。

class Person():    def __init__(self):        print("是一个人")    def eat(self):        print("要吃饭")        Bob = Person()>>>是一个人

【实例2】哪些需放入__ init__ ()方法中,哪些不需要???

需要在 __ init__ ()方法中定义:希望有一些操作是在创建实例的时候就自动创建的。在神经网络代码中,一些网络结构的设置,也最好放在 __ init__ ()方法中。

在下述代码中,我们把money这个属性也定义在__ init__ ()方法中,这样就不需要在执行eat()方法后再执行qian()方法。

class Person():    def __init__(self, name,money):        print("是一个人")        self.name = name        self.money = money    def eat(self):        print("%s要吃饭" % self.name)    def qian(self):        print("花了%s元" % self.money)Bob = Person("Bob",12)Bob.eat()Bob.qian()>>>是一个人Bob要吃饭花了12