我们在写代码的时候往往会遇到这样的问题,几个不同的函数,我们希望它们都会执行某一部分共同的功能,然后再各自执行不同的功能。举个例子,在写神经网络的时候,我们自己定义了很多种层,然后希望在构建网络的同时,把每一层的名字输出来,这个功能怎么实现呢?
方法一:
def conv(...):
print("conv")
...
def fc(...):
print("fc")
...
...
这样在调用的时候会依次输出层名,但这样做实在是费时费力,每一个print()都需要自己手动输入,有没有什么简单的方法呢?
方法二:
def print_info(layer,*args,**kwargs):
print(layer.__name__)
layer(...)
def fc(...):
...
def conv(...):
...
...
在调用的时候,就用print_info(layer)来代替,如调用卷积层的时候print_info(conv),然而这个方法的缺点也很明显,第一非常麻烦,每次都要调用prin_info函数还要传参,第二破坏了原本的代码逻辑。
方法三:
def print_info(layer):
def get_layer(*args, **kwargs):
print(layer.__name__)
layer(...)
return get_layer
def fc(...):
...
def conv(...):
...
在调用的时候,就用func = print_info(func); func(…) 来代替,如调用卷积层的时候conv=print_info(conv); conv(…) 。在这里print_info()就是一个简单的装饰器,它把真正的核心get_layer()函数包装了起来。这样做法的最大好处就是可以在完全不改动原有代码的前提下为函数添加额外功能。但是我们仍然不满意,难道以后堆叠层的时候还每次先得做一个赋值吗?当然不。
方法四:
def print_info(layer):
def get_layer(*args, **kwargs):
print(layer.__name__)
layer(...)
return get_layer
@print_info
def fc(...):
...
@print_info
def conv(...):
...
现在我们可以和以前一样,直接调用conv, fc(),它等价于方法三的写法。这就是我们所谓的语法糖,使原本苦涩的代码结构变得甜蜜起来。