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 (Optional[torch.Tensor]) – 卷積偏置。

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

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

• bn_eps (float) – BatchNorm epsilon。

• bn_w (Optional[torch.Tensor]) – BatchNorm 權重。

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

• transpose (bool, optional) – 如果為 True，則轉置卷積權重。預設為 False。

返回

融合後的卷積權重和偏置。

返回型別

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