Skip to content

Releases: banctilrobitaille/torch-vectorized

v0.1.3

07 Aug 15:34
Compare
Choose a tag to compare

Added documentation

Added eigen values reordering

v0.1.2

23 Jul 15:06
5eb54b2
Compare
Choose a tag to compare

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

22 Feb 19:12
Compare
Choose a tag to compare

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