r"""A fused :class:`~.Module` including :class:`~.module.Linear` and :func:`~.relu`. Could be replaced with :class:`~.QATModule` version :class:`~.qat.LinearRelu` using :func:`~.quantize.quantize_qat`.
| 78 | |
| 79 | |
| 80 | class LinearRelu(Linear): |
| 81 | r"""A fused :class:`~.Module` including :class:`~.module.Linear` and :func:`~.relu`. |
| 82 | Could be replaced with :class:`~.QATModule` version :class:`~.qat.LinearRelu` using :func:`~.quantize.quantize_qat`. |
| 83 | """ |
| 84 | |
| 85 | def forward(self, inp): |
| 86 | return relu(self.calc_linear(inp, self.weight, self.bias)) |