Skip to content

Commit d122b4e

Browse files
bamossoumith
authored andcommitted
Update btrisolve docs to the newest interface.
1 parent 91c4ba7 commit d122b4e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

torch/_torch_docs.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4438,8 +4438,8 @@
44384438
44394439
>>> A = torch.randn(2, 3, 3)
44404440
>>> b = torch.randn(2, 3)
4441-
>>> A_LU_data, A_LU_pivots, info = torch.btrifact(A)
4442-
>>> x = b.trisolve(A_LU_data, A_LU_pivots)
4441+
>>> A_LU = torch.btrifact(A)
4442+
>>> x = b.btrisolve(*A_LU)
44434443
>>> torch.norm(A.bmm(x.unsqueeze(2)) - b)
44444444
6.664001874625056e-08
44454445

0 commit comments

Comments
 (0)