Solve the system of linear equations with strictly upper triangular matrix

1

Now, I have a system of linear equations Ax = y, where A is a strictly upper triangular matrix, I want to solve x efficiently by using the special structure of A. Pay attention to that, in general, this can be solved by x = torch.gesv(y, A), but this function can not utilize the strictly upper triangular structure of A.

1 Like
2

Does torch.trtrs fit your purpose?

Best regards

Thomas

2 Likes
3

Thank you for your reply. This is my need.

4

now I believe it’s being renamed to torch.triangular_solve

2 Likes