torch.nn.utils.fuse_conv_bn_eval

torch.nn.utils.fuse_conv_bn_eval(conv, bn, transpose=False)[源代码]

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

参数

• conv (torch.nn.modules.conv._ConvNd) – 表示一个卷积模块。

• bn (torch.nn.modules.batchnorm._BatchNorm) – 批归一化模块。

• transpose (bool, 可选) – 是否转置卷积权重。默认为 False。

返回值

融合卷积模块

返回类型

_ConvNd (在 torch.nn.modules.conv 模块中)

注意

Both conv and bn must be in evaluation mode, and bn must have its running buffers computed.