Hermitian nd (#293)

* Allow hermitian to take any rank * Fix hermitian Size with ND tensors
NVIDIA · Oct 14, 2022 · f251759 · f251759
1 parent ee70f76
commit f251759
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/include/matx/operators/hermitian.h b/include/matx/operators/hermitian.h
@@ -76,7 +76,8 @@ namespace matx
         }
         constexpr __MATX_INLINE__ __MATX_HOST__ __MATX_DEVICE__ index_t Size(int dim) const
         {
-          return op_.Size(Rank() - dim - 1);
+          // Optimize these branches later
+          return (dim < (Rank() - 2)) ? op_.Size(dim) : op_.Size((dim == Rank() - 1) ? Rank() - 2 : Rank() - 1);
         }
     };
   }