Warning: This API is deprecated and will be removed in a future version of TensorFlow after the replacement is stable.

TridiagonalMatMul

Stay organized with collections Save and categorize content based on your preferences.
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.