pytorch autograd solver

October 8, 2018 · View on GitHub

DONE

symeig
batch_symeig
generalized_symeig

TODO

batch_svd
batch_inv

see torch_autograd_solver_/__init__.py

requirements

torch 0.4.1 - 0.5.0
(CUDA optional) https://github.com/ShigekiKarita/pytorch-cusolver