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.
fast deep-learning pytorch batch matrices analytical analytical-solution 3x3 eigen-decomposition volumetric-3x3-matrices batch-eigen-decomposition
-
Updated
Feb 1, 2021 - Python