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, optional) – 如果為 True，則轉置卷積權重。預設為 False。

返回值

融合後的卷積模組。

返回值型別

torch.nn.modules.conv._ConvNd

注意

conv 和 bn 都必須處於 eval 模式，並且 bn 的執行緩衝區必須已計算。