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

Примечание

И linear, и bn должны находиться в режиме eval, а для bn должны быть вычислены текущие буферы.