I did the same setup as in the example but replace Poincare with Lorentz. This leads to a dimension mismatch and the forward fails.
line 21, in _ldot
m = u * v
RuntimeError: The size of tensor a (3) must match the size of tensor b (4) at non-singleton dimension 0
I did the same setup as in the example but replace Poincare with Lorentz. This leads to a dimension mismatch and the forward fails.