Merge branch 'unifyai:main' into ifftshift

ivy-llc · Apr 9, 2024 · 4ba5781 · 4ba5781
2 parents 1c9df34 + decb87f
commit 4ba5781
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/ivy/functional/frontends/torch/tensor.py b/ivy/functional/frontends/torch/tensor.py
@@ -70,7 +70,7 @@ def dtype(self):
 
     @property
     def shape(self):
-        return Size(ivy.shape(self.ivy_array, as_array=True))
+        return Size(self.ivy_array.shape)
 
     @property
     def real(self):
@@ -1431,7 +1431,12 @@ def bitwise_xor_(self, other):
 
     def item(self):
         if all(dim == 1 for dim in self.shape):
-            return self.ivy_array.to_scalar()
+            if ivy.current_backend_str() == "tensorflow":
+                import tensorflow as tf
+
+                return tf.squeeze(self.ivy_array.data)
+            else:
+                return self.ivy_array.to_scalar()
         else:
             raise ValueError(
                 "only one element tensors can be converted to Python scalars"