Added l2norm layer to nnj

MachineLearningLifeScience / stochman

Algorithms for computations on random manifolds made easier

Apache License 2.0

85 stars 11 forks source link

Closed FrederikWarburg closed 2 years ago

FrederikWarburg commented 2 years ago

Hi Nicki

We added a l2norm layer to nnj. It follows same setup as other functions, except it does not inherit from torch.nn. We tested in a separate notebook.