lstsq

ivy-llc · Aug 25, 2023 · 3fb83e5 · 3fb83e5
1 parent 8669bc9
commit 3fb83e5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/ivy/functional/frontends/torch/linalg.py b/ivy/functional/frontends/torch/linalg.py
@@ -26,6 +26,9 @@ def diagonal(A, *, offset=0, dim1=-2, dim2=-1):
 def divide(input, other, *, rounding_mode=None, out=None):
     return ivy.divide(input, other, out=out)
 
+@to_ivy_arrays_and_back
+def lstsq(A, B, rcond=None, *, driver=None):
+    return torch_frontend.lstsq(A, B, rcond=rcond, driver=driver)
 
 @to_ivy_arrays_and_back
 @with_supported_dtypes(