快捷方式

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 模块。

返回值

融合后的线性模块。

返回类型

torch.nn.Linear

注意

linearbn 都必须处于评估模式,并且 bn 必须已计算其运行时缓冲区。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源