快捷方式

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, 可选) – 如果为 True,则转置卷积权重。默认为 False。

返回值

融合后的卷积模块。

返回类型

torch.nn.modules.conv._ConvNd

注意

convbn 必须都处于评估模式,并且 bn 必须计算其运行缓冲区。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并解答您的问题

查看资源