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.