fix: more elegant implementation of ivy.Shape.numel

ivy-llc · Jul 10, 2024 · 8a93b6d · 8a93b6d
1 parent ae17824
commit 8a93b6d
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/ivy/__init__.py b/ivy/__init__.py
@@ -493,9 +493,7 @@ def numel(self):
             raise ivy.utils.exceptions.IvyException(
                 "Cannot calculate the number of elements in a partially known Shape"
             )
-        if self.rank > 0:
-            return ivy.prod(self._shape, dtype=ivy.int64).to_scalar()
-        return 1
+        return ivy.prod(self.as_list(), dtype=ivy.int64).to_scalar()
 
     def __concat__(self, other):
         return self.concatenate(other)