Solve linear system with Variable matrix

1

Hello!
I have to solve linear system with Variable matrix and then to compute gradients with respect to it.
Is there any way to perform this operation?

1 Like
2

I think you’re looking for torch.gesv

3

Thanks a lot, it was rather hard to find this function!

1 Like
4

gesv can only support the matrix A being square. Is it possible to solve linear system for non-square A in PyTorch?

5

Linear system for non-square is an overparametrized linar system which is solved by minimization, typically svd. Im not sure if it’s differentiable

1 Like
6

For non-square A, you should look at gels, not gesv. https://pytorch.org/docs/stable/torch.html#torch.gels