gels -> lstsq

Signed-off-by: Edward Z. Yang <[email protected]>

Author: ezyang

torchvision/transforms/functional.py

@@ -437,7 +437,7 @@ def _get_perspective_coeffs(startpoints, endpoints):
 
     A = torch.tensor(matrix, dtype=torch.float)
     B = torch.tensor(startpoints, dtype=torch.float).view(8)
-    res = torch.gels(B, A)[0]
+    res = torch.lstsq(B, A)[0]
     return res.squeeze_(1).tolist()