Return 0D arrays for 0D shape in operators

NVIDIA · Jan 12, 2024 · 34292e9 · 34292e9
1 parent 43ea5b8
commit 34292e9
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/include/matx/operators/base_operator.h b/include/matx/operators/base_operator.h
@@ -173,8 +173,7 @@ namespace matx
 
         __MATX_INLINE__ __MATX_HOST__ __MATX_DEVICE__ auto Shape() const {
           if constexpr (T::Rank() == 0) {
-            std::array<index_t, 1> sizes_{1};
-            return sizes_;
+            return std::array<index_t, 0> {};
           }
           else {
             std::array<index_t, T::Rank()> sizes_;