Fast analytical implementation of batch eigen-decomposition for 3x3 symmetric matrices with Pytorch. > 250x faster than regular Pytorch implementation of batch eigen-decomposition on GPU.
Home Page:https://torch-vectorized.readthedocs.io/en/latest/
Geek Repo:Geek Repo
Github PK Tool:Github PK Tool