Skip to content

Releases: banctilrobitaille/torch-vectorized

v0.1.3

Choose a tag to compare

@banctilrobitaille banctilrobitaille released this 07 Aug 15:34

Added documentation

Added eigen values reordering

v0.1.2

Choose a tag to compare

@banctilrobitaille banctilrobitaille released this 23 Jul 15:06
5eb54b2

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

Choose a tag to compare

@banctilrobitaille banctilrobitaille released this 22 Feb 19:12

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