paddle.numel tensor instance function (ivy-llc#16614)

mobley-trent · Aug 1, 2023 · d20f522 · d20f522
1 parent 5499bbf
commit d20f522
Show file tree

Hide file tree

Showing 3 changed files with 126 additions and 541 deletions.
diff --git a/ivy/functional/frontends/paddle/tensor/stat.py b/ivy/functional/frontends/paddle/tensor/stat.py
@@ -22,7 +22,7 @@ def numel(x, name=None):
         length = len(x)
     except (ValueError, TypeError):
         length = 1  # if 0 dimensional tensor with 1 element
-    return ivy.array([prod if prod > 0 else ivy.array(length, dtype=ivy.int64)])
+    return ivy.array(prod if prod > 0 else ivy.array(length, dtype=ivy.int64))
 
 
 @with_unsupported_dtypes({"2.4.2 and below": ("float16", "bfloat16")}, "paddle")

diff --git a/ivy/functional/frontends/paddle/tensor/tensor.py b/ivy/functional/frontends/paddle/tensor/tensor.py
@@ -108,6 +108,10 @@ def acosh(self, name=None):
     def ceil(self):
         return paddle_frontend.ceil(self)
 
+    @with_unsupported_dtypes({"2.5.1 and below": ("complex", "int8")}, "paddle")
+    def numel(self):
+        return paddle_frontend.numel(self)
+
     @with_unsupported_dtypes({"2.5.1 and below": ("float16",)}, "paddle")
     def asinh(self, name=None):
         return paddle_frontend.Tensor(ivy.asinh(self._ivy_array))