Question about torch.lstsq and torch.linalg.lstsq #128952
Labels
module: linear algebra
Issues related to specialized linear algebra operations in PyTorch; includes matrix multiply matmul
triaged
This issue has been looked at a team member, and triaged and prioritized into an appropriate module
My pytorch version is 1.12.1.
I used
torch.lstsq(A, B)[0][B.shape[1]]
andtorch.linalg.lstsq(B, A).solution
and the results are different. I want to get the answer of the former. How should I usetorch.linalg.lstsq
to get it?cc @jianyuh @nikitaved @pearu @mruberry @walterddr @xwang233 @lezcano
The text was updated successfully, but these errors were encountered: