There was an error while loading. Please reload this page.
1 parent 91c4ba7 commit d122b4eCopy full SHA for d122b4e
torch/_torch_docs.py
@@ -4438,8 +4438,8 @@
4438
4439
>>> A = torch.randn(2, 3, 3)
4440
>>> 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)
+ >>> A_LU = torch.btrifact(A)
+ >>> x = b.btrisolve(*A_LU)
4443
>>> torch.norm(A.bmm(x.unsqueeze(2)) - b)
4444
6.664001874625056e-08
4445
0 commit comments