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) – 一个 BatchNorm 模块。
transpose (bool, optional) – 如果为 True，则转置卷积权重。默认为 False。
- 返回值
融合后的卷积模块。
- 返回值类型
torch.nn.modules.conv._ConvNd
注意
conv和bn都必须处于 eval 模式，并且bn的运行缓冲区必须已计算。
