快捷方式

torch.nn.utils.fuse_conv_bn_weights

torch.nn.utils.fuse_conv_bn_weights(conv_w, conv_b, bn_rm, bn_rv, bn_eps, bn_w, bn_b, transpose=False)[源代码]

将卷积模块参数和 BatchNorm 模块参数融合成新的卷积模块参数。

参数
  • conv_w (torch.Tensor) – 卷积权重。

  • conv_b (可选[torch.Tensor]) – 卷积偏置。

  • bn_rm (torch.Tensor) – BatchNorm 运行均值。

  • bn_rv (torch.Tensor) – BatchNorm 运行方差。

  • bn_eps (float) – BatchNorm ϵ。

  • bn_w (可选[torch.Tensor]) – BatchNorm 权重。

  • bn_b (可选[torch.Tensor]) – BatchNorm 偏置。

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

返回

融合后的卷积权重和偏置。

返回类型

Tuple[torch.nn.Parameter, torch.nn.Parameter]

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获取问题解答

查看资源