MachineLearningLifeScience / stochman

Algorithms for computations on random manifolds made easier
Apache License 2.0
85 stars 11 forks source link

Manifold.curve_energy does not support batching #15

Open ciliew opened 2 years ago

ciliew commented 2 years ago

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