Releases: banctilrobitaille/torch-vectorized
Releases · banctilrobitaille/torch-vectorized
v0.1.3
v0.1.2
Added determinant and trace operation as vDet and vTrace.
Added support for empty matrices and same diag matrices
Added nans replacement in vExpm and vLogm.
Initial release
First realease of a quick closed-form solution for volumetric 3x3 matrices Eigen-Decomposition with Pytorch. Solves Eigen-Decomposition of data with shape Bx9xDxHxW, where B is the batch size, 9 is the flattened 3x3 symmetric matrices, D is the depth, H is the Height, W is the width.
Features:
- Decomposition of batched volume of 3x3 symmetric matrices of shape (B,9,D,H,W) using a analytical solution with Pytorch
- Compute or not the eigenvectors using the eigen_vectors flag
- Return the eigen-values and eigen-vectors with shape (B, 3, D, H, W)/(B, 3,3,D,H,W) or flattened (BxDxHxW, 3)/(BxDxHxW, 3, 3) with the flattened_output flag