Error while computing trajectory matrix

[UriKH]

The bug: When computing a trajectory matrix for a valid trajectory which doesn't step out of the shard it's in, we get a zero-division error.

Example: see the Colab link.

Note: As discussed, this is an issue with the order we multiply matrices and substitute when creating the trajectory matrix.