torch.nn.utils.fuse_linear_bn_eval

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

將一個線性模組 (linear module) 和一個 BatchNorm 模組融合成一個新的線性模組。

引數

• linear (torch.nn.Linear) – 一個線性模組。

• bn (torch.nn.modules.batchnorm._BatchNorm) – 一個 BatchNorm 模組。

返回值

融合後的線性模組。

返回型別

torch.nn.Linear

注意

linear 和 bn 都必須處於評估模式 (eval mode)，並且 bn 的執行緩衝區 (running buffers) 必須已計算。