Attend the Women in ML Symposium on December 7 Register now

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.