torch.nn.utils.fuse_linear_bn_eval

torch.nn.utils.fuse_linear_bn_eval(linear, bn)

将一个线性模块和一个批归一化模块融合成一个单一的新的线性模块。

参数:

• 线性（torch.nn.Linear）- 线性模块。

• bn（torch.nn.modules.batchnorm._BatchNorm）- 批归一化模块。

返回值:

合并的线性模块。

返回类型:

torch.nn.Linear

注意

linear 和 bn 必须处于评估模式，且 bn 必须计算其运行缓冲区。