亲宝软件园·资讯

展开

Python super(XXXX, self).__init__()

波尔德 人气:0

在使用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元

加载全部内容

相关教程
猜你喜欢
用户评论