{ }
Solves tridiagonal systems of equations.
tf.raw_ops.TridiagonalSolve(
diagonals, rhs, partial_pivoting=True, perturb_singular=False, name=None
)
Solves tridiagonal systems of equations.
Supports batch dimensions and multiple right-hand sides per each left-hand
side.
On CPU, solution is computed via Gaussian elimination with or without partial
pivoting, depending on partial_pivoting
attribute. On GPU, Nvidia's cuSPARSE
library is used: https://docs.nvidia.com/cuda/cusparse/index.html#gtsv
Partial pivoting is not yet supported by XLA backends.
Returns | |
---|---|
A Tensor . Has the same type as diagonals .
|