torch.nn.utils.fuse_linear_bn_eval¶
- torch.nn.utils.fuse_linear_bn_eval(linear, bn)[源代码]¶
将线性模块和 BatchNorm 模块融合到一个新的线性模块中。
- 参数
linear (torch.nn.Linear) – 线性模块。
bn (torch.nn.modules.batchnorm._BatchNorm) – BatchNorm 模块。
- 返回值
融合后的线性模块。
- 返回类型
注意
linear
和bn
都必须处于评估模式,并且bn
必须已计算其运行时缓冲区。