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)[source][source]¶
将卷积模块参数和 BatchNorm 模块参数融合成新的卷积模块参数。
- 参数:
conv_w (torch.Tensor) – 卷积权重。
conv_b (Optional[torch.Tensor]) – 卷积偏差。
bn_rm (torch.Tensor) – BatchNorm 运行均值。
bn_rv (torch.Tensor) – BatchNorm 运行方差。
bn_eps (浮点数) – 批标准化 epsilon。
bn_w (Optional[torch.Tensor]) – 批标准化权重。
bn_b (Optional[torch.Tensor]) – 批标准化偏差。
transpose (bool, 可选) – 如果为 True，则转置卷积权重。默认为 False。
- 返回值:
融合卷积权重和偏置。
- 返回类型:
元组[torch.nn.Parameter, torch.nn.Parameter]
