torch.nn.utils.fuse_conv_bn_eval

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

将卷积模块和批量归一化模块融合成一个单一的新卷积模块。

Parameters

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

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

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

Returns

融合的卷积模块。

Return type

torch.nn.modules.conv._ConvNd

注意

Both conv and bn 必须处于评估模式，并且 bn 必须计算其运行缓冲区。