Giters

ksteensig / pytorch-batch-ops

PyTorch batch symeig and (low-rank) svd implemented using cuSOLVER

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

PyTorch batch operations

PyTorch batch symeig and (low-rank) svd implemented using cuSOLVER

About

PyTorch batch symeig and (low-rank) svd implemented using cuSOLVER

License:MIT License

Languages

Language:Python 50.6%Language:C++ 49.4%