Solve the system of linear equations with strictly upper triangular matrix

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

Does torch.trtrs fit your purpose?

Best regards



Thank you for your reply. This is my need.

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