__call__、__init__和 forward三个函数在python中的调用方法

这三个函数包括了Python中特有的重载函数,以及神经网络中向前传播的重要函数。这是我看到的比较清楚的说明Python中这三个函数功能的博客。
转载地址:博客引入来源

init: 类的初始化函数,类似于c++的构造函数
call_: 使得类对象具有类似函数的功能。
__init__比较好理解,现在主要看一下 __call__的功能示例:

class A():
def call(self):
print(‘i can be called like a function’)

a = A()
a()
out:

i can be called like a function

让我们在调用时传入参数如何?

class A():
def call(self, param):

    print('i can called like a function')
    print('掺入参数的类型是:', type(param))

a = A()

a(‘i’)
out:

i can called like a function
掺入参数的类型是: <class ‘str’>

发现对象a的表现完全类似一个函数。

那当然也可以在__call__里调用其他的函数啊,
在__call__函数中调用forward函数,并且返回调用的结果

class A():
def call(self, param):

    print('i can called like a function')
    print('传入参数的类型是:{}   值为: {}'.format(type(param), param))

    res = self.forward(param)
    return res

def forward(self, input_):
    print('forward 函数被调用了')

    print('in  forward, 传入参数类型是:{}  值为: {}'.format( type(input_), input_))
    return input_

a = A()

input_param = a(‘i’)
print(“对象a传入的参数是:”, input_param)

out:

i can called like a function
传入参数的类型是:<class ‘str’> 值为: i
forward 函数被调用了
in forward, 传入参数类型是:<class ‘str’> 值为: i
对象a传入的参数是: i

现在我们将初始化函数__init__也加上,来看一下:
在对象初始化时确定初始年龄,通过调用a(2)为对象年龄增加2岁,

class A():
def init(self, init_age):
super().init()
print(‘我年龄是:’,init_age)
self.age = init_age

def __call__(self, added_age):
    

    res = self.forward(added_age)
    return res

def forward(self, input_):
    print('forward 函数被调用了')
    
    return input_ + self.age

print(‘对象初始化。。。。’)
a = A(10)

input_param = a(2)
print(“我现在的年龄是:”, input_param)

out:

对象初始化。。。。
我年龄是: 10
forward 函数被调用了
我现在的年龄是: 12

pytorch主要也是按照__call__, init,forward三个函数实现网络层之间的架构的

这博客讲述了pytorch中具体实现:https://blog.csdn.net/dss_dssssd/article/details/82977170

上一篇:01|分形树绘制-Turtle库基础知识


下一篇:Angular和Spring Boot一起做个项目