torch.nn.utils.fusion.fuse_linear_bn_eval

torch.nn.utils.fusion.fuse_linear_bn_eval(linear, bn) [исходный код]

Объединяет модуль Linear и модуль BatchNorm в один новый модуль Linear.

Параметры:

• linear (torch.nn.Linear) – Модуль Linear.
• bn (torch.nn.modules.batchnorm._BatchNorm) – Модуль BatchNorm.

Возвращает:

Объединённый модуль Linear.

Тип возвращаемого значения:

torch.nn.Linear

Примечание

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