def forward_diff(self, d, u, v):
        """Computes the 'forward diff' or Euler update rule: (I - d A)^-1 u + d B v.

        d: (...)
        u: (..., n)
        v: (...)
        """

        v = d * v
        v = v.unsqueeze(-1) * self.B
        x = self.forward_mult(u, d)
        x = x + v
        return x
Computes the 'forward diff' or Euler update rule: (I - d A)^-1 u + d B v.

        d: (...)
        u: (..., n)
        v: (...)
        
analyze/convnexts4nd/src/models/hippo/transition.py