Rate this Page

torch.nn.utils.fuse_linear_bn_weights#

torch.nn.utils.fuse_linear_bn_weights(linear_w, linear_b, bn_rm, bn_rv, bn_eps, bn_w, bn_b)[source]#

Fuse linear module parameters and BatchNorm module parameters into new linear module parameters.

Parameters
Returns

Fused linear weight and bias.

Return type

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