torch.nn.utils.fuse_conv_bn_eval¶
- torch.nn.utils.fuse_conv_bn_eval(conv, bn, transpose=False)[source][source]¶
将卷积模块和批归一化模块融合成一个单一的新的卷积模块。
- 参数:
conv (torch.nn.modules.conv._ConvNd) – 一个卷积模块。
bn (torch.nn.modules.batchnorm._BatchNorm) – 一个批归一化模块。
transpose (bool, 可选) – 如果为 True，则转置卷积权重。默认为 False。
- 返回值:
合并的卷积模块。
- 返回类型:
torch.nn.modules.conv._ConvNd
注意
两个
conv和bn必须处于评估模式，且bn必须计算其运行缓冲区。
