fix: Make torch.pow frontend accept scalar input

ivy-llc · Sep 12, 2024 · fe60be9 · fe60be9
1 parent c11aeee
commit fe60be9
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/ivy/functional/frontends/torch/pointwise_ops.py b/ivy/functional/frontends/torch/pointwise_ops.py
@@ -411,6 +411,10 @@ def positive(input, *, out=None):
 @with_unsupported_dtypes({"2.2 and below": ("bool",)}, "torch")
 @to_ivy_arrays_and_back
 def pow(input, exponent, *, out=None):
+    # torch supports input not being a tensor as well
+    if not ivy.is_array(input):
+        input = torch_frontend.as_tensor(input).ivy_array
+
     if not ivy.is_array(exponent):
         if (
             any(dtype in str(input.dtype) for dtype in ["int8", "int16"])