Rate this Page
★★★★★

torch.nn.utils.fuse_linear_bn_eval#

torch.nn.utils.fuse_linear_bn_eval(linear,bn)[source]#

Fuse a linear module and a BatchNorm module into a single, new linear module.

Parameters

  • linear (torch.nn.Linear) – A Linear module.

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

Returns

The fused linear module.

Return type

torch.nn.Linear

Note

Bothlinear andbn must be in eval mode, andbn must have its running buffers computed.