在使用pytorch框架时,难免要自己定义网络。于是,super(XXXX, self).init(),就成了自定义网络结构时必不可少的第一句。但是,super(XXXX, self).init()具体的作用是什么我一直没有搞清楚。阅读了大量的博客后,我终于搞懂了! 一言以蔽之:super(XXX, self).init()——对继承自父类的属性进行初始化,并且用父类的初始化方法初始化继承的属性。 我们先看一个简单的例子:
class Person():
def __init__(self, name, gender):
self.name = name
self.gender = gender
def printinfo(self):
print(self.name, self.gender)
class Stu(Person):
def __init__(self, name, gender, school):
super(Stu, self).__init__(name, gender)
self.school = school
def printinfo(self):
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
class Student(Person):
def __init__(self, name, gender, age, school, score):
super(Student, self).__init__(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):
def __init__(self):
super(Net, self).__init__()
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元
|