ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

TridiagonalMatMul

public final class TridiagonalMatMul

Calculate product with tridiagonal matrix.

Calculates product of two matrices, where left matrix is a tridiagonal matrix.

Public Methods

Output<T>
asOutput()
Returns the symbolic handle of a tensor.
static <T> TridiagonalMatMul<T>
create(Scope scope, Operand<T> superdiag, Operand<T> maindiag, Operand<T> subdiag, Operand<T> rhs)
Factory method to create a class wrapping a new TridiagonalMatMul operation.
Output<T>
output()
Tensor of shape `[..., M, N]` containing the product.

Inherited Methods

Public Methods

public Output<T> asOutput ()

Returns the symbolic handle of a tensor.

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.

public static TridiagonalMatMul<T> create (Scope scope, Operand<T> superdiag, Operand<T> maindiag, Operand<T> subdiag, Operand<T> rhs)

Factory method to create a class wrapping a new TridiagonalMatMul operation.

Parameters
scope current scope
superdiag Tensor of shape `[..., 1, M]`, representing superdiagonals of tri-diagonal matrices to the left of multiplication. Last element is ignored.
maindiag Tensor of shape `[..., 1, M]`, representing main diagonals of tri-diagonal matrices to the left of multiplication.
subdiag Tensor of shape `[..., 1, M]`, representing subdiagonals of tri-diagonal matrices to the left of multiplication. First element is ignored.
rhs Tensor of shape `[..., M, N]`, representing MxN matrices to the right of multiplication.
Returns
  • a new instance of TridiagonalMatMul

public Output<T> output ()

Tensor of shape `[..., M, N]` containing the product.