functools 中的 partial()函数的作用

参考博文链接

1、函数形式

functools.partial(func[,*args][, **kwargs])

2、例子

假设有如下函数：

def multiply(x, y):
    return x * y

现在，我们想返回某个数的双倍，即：

>>> multiply(3, y=2)
6
>>> multiply(4, y=2)
8
>>> multiply(5, y=2)
10

上面的调用有点繁琐，每次都要传入 y=2，我们想到可以定义一个新的函数，把 y=2 作为默认值，即：

def double(x, y=2):
    return multiply(x, y)

现在，我们可以这样调用了：

>>> double(3)
6
>>> double(4)
8
>>> double(5)
10

事实上，我们可以不用自己定义 double，利用 partial，我们可以这样：

from functools import partial
 
double = partial(multiply, y=2)

3、总结

partial 接收函数 multiply 作为参数，固定 multiply 的参数 y=2，并返回一个新的函数给 double，这跟我们自己定义 double 函数的效果是一样的。所以，简单而言，partial 函数的功能就是：把一个函数的某些参数给固定住，返回一个新的函数。需要注意的是，我们上面是固定了 multiply 的关键字参数 y=2。

# partial(): 把nn.BatchNorm2d函数中的eps、momentum两个参数固定住，返回一个新的函数norm_layer
norm_layer = partial(nn.BatchNorm2d, eps=1e-3, momentum=0.1)