torch.nn.utils.fusion.fuse_conv_bn_weights

torch.nn.utils.fusion.fuse_conv_bn_weights(conv_w, conv_b, bn_rm, bn_rv, bn_eps, bn_w, bn_b, transpose=False) [исходный код]

Объединяет параметры свёрточного модуля и модуля BatchNorm в параметры нового свёрточного модуля.

Параметры:

• conv_w (torch.Tensor) – Вес свёрточного слоя.
• conv_b (Optional[torch.Tensor]) – Смещение свёрточного слоя.
• bn_rm (torch.Tensor) – Текущее среднее BatchNorm.
• bn_rv (torch.Tensor) – Текущая дисперсия BatchNorm.
• bn_eps (float) – Эпсилон BatchNorm.
• bn_w (Optional[torch.Tensor]) – Вес BatchNorm.
• bn_b (Optional[torch.Tensor]) – Смещение BatchNorm.
• transpose (bool, необязательный) – Если True, транспонировать вес свёрточного слоя. По умолчанию False.

Возвращает:

Объединённые вес и смещение свёрточного слоя.

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

Tuple[torch.nn.Parameter, torch.nn.Parameter]