[DOCS]: Correct docs for torch.lu_solve

[bharatr21]

Fixes #43498 by correcting the function signature of torch.lu_solve

[dr-ci]

💊 CI failures summary and remediations

As of commit 9037d65 (more details on the Dr. CI page):

---

• 1/4 failures possibly* introduced in this PR
  • 1/1 non-CircleCI failure(s)
• 3/4 broken upstream at merge base 88ec72e since Nov 11

---

🚧 3 ongoing upstream failures:

These were probably caused by upstream breakages that are not fixed yet:

• pytorch_xla_linux_bionic_py3_6_clang9_test since Nov 10
  • 🔁 rerun
• pytorch_linux_xenial_cuda10_2_cudnn7_py3_gcc7_test2 since Nov 11
  • 🔁 rerun
• pytorch_linux_xenial_cuda10_2_cudnn7_py3_gcc7_test1 since Nov 11
  • 🔁 rerun

---

ci.pytorch.org: 1 failed

• Failed: pr/pytorch-linux-bionic-rocm3.9-py3.6

---

This comment was automatically generated by Dr. CI (expand for details).

Follow this link to opt-out of these comments for your Pull Requests.

Please report bugs/suggestions on the GitHub issue tracker or post in the (internal) Dr. CI Users group.

See how this bot performed.

This comment has been revised 3 times.

[facebook-github-bot]

@ailzhang has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.

[facebook-github-bot]

@ailzhang merged this pull request in 3aeb9cc.

[kshitij12345]

I think,
The input argument b in the docs should be changed to input.

>>> A = torch.randn(2, 3, 3)
>>> b = torch.randn(2, 3, 1)
>>> A_LU = torch.lu(A)
>>> x = torch.lu_solve(b, *A_LU)
>>> torch.lu_solve(input=b, LU_data=A_LU[0], LU_pivots=A_LU[1])
tensor([[[ 2.0788],
         [ 0.1862],
         [ 2.7321]],

        [[ 1.7038],
         [ 1.7400],
         [-3.2406]]])
>>> torch.lu_solve(b=b, LU_data=A_LU[0], LU_pivots=A_LU[1])
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: lu_solve() missing 3 required positional argument: "input", "LU_data", "LU_pivots"