noahgolmant / pytorch-hessian-eigenthings

Efficient PyTorch Hessian eigendecomposition tools!
MIT License
360 stars 43 forks source link

Fix eigenvalue to be a float instead of a tensor during power iteration #9

Closed noahgolmant closed 5 years ago

noahgolmant commented 5 years ago

Resolves #8