r"""A fused :class:`~.Module` including :class:`~.module.Conv2d` and :class:`~.module.BatchNorm2d`. Could be replaced with :class:`~.QATModule` version :class:`~.qat.ConvBn2d` using :func:`~.quantize.quantize_qat`.
| 45 | |
| 46 | |
| 47 | class ConvBn2d(_ConvBnActivation2d): |
| 48 | r"""A fused :class:`~.Module` including :class:`~.module.Conv2d` and :class:`~.module.BatchNorm2d`. |
| 49 | Could be replaced with :class:`~.QATModule` version :class:`~.qat.ConvBn2d` using |
| 50 | :func:`~.quantize.quantize_qat`. |
| 51 | """ |
| 52 | |
| 53 | def forward(self, inp): |
| 54 | return self.bn(self.conv(inp)) |
| 55 | |
| 56 | |
| 57 | class ConvBnRelu2d(_ConvBnActivation2d): |