Manifold.curve_energy should have a dim argument in the return statement, ie. line 47 ` return energy.sum()` should be `return energy.sum(dim= ... )`.