torch (version 0.2.0)

torch_triangular_solve: Triangular_solve

Description

Triangular_solve

Usage

torch_triangular_solve(
  self,
  A,
  upper = TRUE,
  transpose = FALSE,
  unitriangular = FALSE
)

Arguments

self

(Tensor) multiple right-hand sides of size $(\ast ,m,k)$ where $\ast$ is zero of more batch dimensions ($b$)

A

(Tensor) the input triangular coefficient matrix of size $(\ast ,m,m)$ where $\ast$ is zero or more batch dimensions

upper

(bool, optional) whether to solve the upper-triangular system of equations (default) or the lower-triangular system of equations. Default: TRUE.

transpose

(bool, optional) whether $A$ should be transposed before being sent into the solver. Default: FALSE.

unitriangular

(bool, optional) whether $A$ is unit triangular. If TRUE, the diagonal elements of $A$ are assumed to be 1 and not referenced from $A$. Default: FALSE.

triangular_solve(input, A, upper=TRUE, transpose=False, unitriangular=False) -> (Tensor, Tensor)

Solves a system of equations with a triangular coefficient matrix $A$ and multiple right-hand sides $b$.

In particular, solves $AX=b$ and assumes $A$ is upper-triangular with the default keyword arguments.

torch_triangular_solve(b, A) can take in 2D inputs b, A or inputs that are batches of 2D matrices. If the inputs are batches, then returns batched outputs X

Examples

# NOT RUN {
if (torch_is_installed()) {

A = torch_randn(c(2, 2))$triu()
A
b = torch_randn(c(2, 3))
b
torch_triangular_solve(b, A)
}
# }