T
- data type for output()
outputpublic final class TridiagonalSolve<T> extends PrimitiveOp implements Operand<T>
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
Modifier and Type | Class and Description |
---|---|
static class |
TridiagonalSolve.Options
Optional attributes for
TridiagonalSolve |
operation
Modifier and Type | Method and Description |
---|---|
Output<T> |
asOutput()
Returns the symbolic handle of a tensor.
|
static <T> TridiagonalSolve<T> |
create(Scope scope,
Operand<T> diagonals,
Operand<T> rhs,
TridiagonalSolve.Options... options)
Factory method to create a class wrapping a new TridiagonalSolve operation.
|
Output<T> |
output()
Tensor of shape `[..., M, K]` containing the solutions
|
static TridiagonalSolve.Options |
partialPivoting(Boolean partialPivoting) |
equals, hashCode, op, toString
public static <T> TridiagonalSolve<T> create(Scope scope, Operand<T> diagonals, Operand<T> rhs, TridiagonalSolve.Options... options)
scope
- current scopediagonals
- Tensor of shape `[..., 3, M]` whose innermost 2 dimensions represent the
tridiagonal matrices with three rows being the superdiagonal, diagonals, and
subdiagonals, in order. The last element of the superdiagonal and the first
element of the subdiagonal is ignored.rhs
- Tensor of shape `[..., M, K]`, representing K right-hand sides per each
left-hand side.options
- carries optional attributes valuespublic static TridiagonalSolve.Options partialPivoting(Boolean partialPivoting)
partialPivoting
- Whether to apply partial pivoting. Partial pivoting makes the procedure more
stable, but slower.public Output<T> asOutput()
Operand
Inputs to TensorFlow operations are outputs of another TensorFlow operation. This method is used to obtain a symbolic handle that represents the computation of the input.
asOutput
in interface Operand<T>
OperationBuilder.addInput(Output)
Copyright © 2022. All rights reserved.