def _A(self):
        """The matrix A for system x' = -Ax + Bu."""

        A = trid_solve(self.I, self.dl, self.d, self.du).transpose(-1, -2)
        A = A + self.c*self.I
        A = self.pl.unsqueeze(-1) * A * self.pr
        return A
The matrix A for system x' = -Ax + Bu.
analyze/convnexts4nd/src/models/hippo/transition.py