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