torch.nn.utils.fusion.fuse_conv_bn_eval

torch.nn.utils.fusion.fuse_conv_bn_eval(conv, bn, transpose=False)

将卷积模块和 BatchNorm 模块融合为一个新的卷积模块。

参数

• conv (torch.nn.modules.conv._ConvNd) – 卷积模块。

• bn (torch.nn.modules.batchnorm._BatchNorm) – BatchNorm 模块。

• transpose (bool, optional) – 如果为 True，则转置卷积权重。默认为 False。

返回

融合后的卷积模块。

返回类型

torch.nn.modules.conv._ConvNd

注意

conv 和 bn 都必须处于 eval 模式，并且 bn 必须已计算其 running buffers。