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)[源代码]

将卷积模块参数与批处理规范化模块参数融合成新的卷积模块参数。

参数

• conv_w (torch.Tensor) – 表示卷积操作的权重。

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

• bn_rm (torch.Tensor) – 批归一化运行均值。

• bn_rv (torch.Tensor) – 批归一化运行方差。

• bn_eps (float) – BatchNorm 的 epsilon 值。

• bn_w (Optional[torch.Tensor]) – 可选的批处理规范化权重。

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

• transpose (bool, 可选) – 是否将卷积权重转置。默认值为 False。

返回值

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

返回类型

(torch.nn.Parameter, torch.nn.Parameter)