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