stochman
stochman copied to clipboard
Manifold.curve_energy does not support batching
Manifold.curve_energy should have a dim argument in the return statement, ie. line 47 return energy.sum() should be return energy.sum(dim= ... ).