banctilrobitaille/torch-vectorized

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.

Language: Python - Size: 396 KB - Last synced at: about 1 year ago - Pushed at: over 4 years ago - Stars: 20 - Forks: 3