torch.nn.utils.fuse_conv_bn_eval

torch.nn.utils.fuse_conv_bn_eval(conv, bn, transpose=False)

Fuse a convolutional module and a BatchNorm module into a single, new convolutional module。

参数

• conv (torch.nn.modules.conv._ConvNd) – A convolutional module。

• bn (torch.nn.modules.batchnorm._BatchNorm) – A BatchNorm module。

• transpose (bool, optional) – If True, transpose the convolutional weight。Defaults to False。

返回

The fused convolutional module。

返回类型

torch.nn.modules.conv._ConvNd

注意

Both conv and bn must be in eval mode, and bn must have its running buffers computed。