def forward(self, x: torch.Tensor) -> torch.Tensor:
        """Apply residual connection to any sublayer with the same size."""
        x_norm = self.norm(x)

        out = self.fn_ff(x_norm)

        return x + self.dropout(out)
Apply residual connection to any sublayer with the same size.
braindecode/models/sleep_stager_eldele_2021.py