linalg_solve: Computes the solution of a square system of linear equations...

In torch: Tensors and Neural Networks with 'GPU' Acceleration

Computes the solution of a square system of linear equations with a unique solution.

Description

Letting \teqn\mathbbK be \teqn\mathbbR or \teqn\mathbbC, this function computes the solution \teqnX \in \mathbbK^n \times k of the linear system associated to \teqnA \in \mathbbK^n \times n, B \in \mathbbK^m \times k, which is defined as

Usage

linalg_solve(A, B)

Arguments

A	(Tensor): tensor of shape ⁠(*, n, n)⁠ where * is zero or more batch dimensions.
B	(Tensor): right-hand side tensor of shape ⁠(*, n)⁠ or ⁠(*, n, k)⁠ or ⁠(n,)⁠ or ⁠(n, k)⁠ according to the rules described above

Details

AX = B

This system of linear equations has one solution if and only if \teqnA is invertible_. This function assumes that \teqnA is invertible. Supports inputs of float, double, cfloat and cdouble dtypes. Also supports batches of matrices, and if the inputs are batches of matrices then the output has the same batch dimensions.

Letting * be zero or more batch dimensions,

• If A has shape ⁠(*, n, n)⁠ and B has shape ⁠(*, n)⁠ (a batch of vectors) or shape ⁠(*, n, k)⁠ (a batch of matrices or "multiple right-hand sides"), this function returns X of shape ⁠(*, n)⁠ or ⁠(*, n, k)⁠ respectively.

• Otherwise, if A has shape ⁠(*, n, n)⁠ and B has shape ⁠(n,)⁠ or ⁠(n, k)⁠, B is broadcasted to have shape ⁠(*, n)⁠ or ⁠(*, n, k)⁠ respectively.

This function then returns the solution of the resulting batch of systems of linear equations.

Note

This function computes X = A$inverse() @ B in a faster and more numerically stable way than performing the computations separately.

See Also

Other linalg: linalg_cholesky_ex(), linalg_cholesky(), linalg_det(), linalg_eigh(), linalg_eigvalsh(), linalg_eigvals(), linalg_eig(), linalg_householder_product(), linalg_inv_ex(), linalg_inv(), linalg_lstsq(), linalg_matrix_norm(), linalg_matrix_power(), linalg_matrix_rank(), linalg_multi_dot(), linalg_norm(), linalg_pinv(), linalg_qr(), linalg_slogdet(), linalg_solve_triangular(), linalg_svdvals(), linalg_svd(), linalg_tensorinv(), linalg_tensorsolve(), linalg_vector_norm()

Examples

if (torch_is_installed()) {
A <- torch_randn(3, 3)
b <- torch_randn(3)
x <- linalg_solve(A, b)
torch_allclose(torch_matmul(A, x), b)
}

torch documentation built on May 29, 2024, 9:54 a.m.