Calculate product with tridiagonal matrix.
tf.raw_ops.TridiagonalMatMul(
superdiag, maindiag, subdiag, rhs, name=None
)
Calculates product of two matrices, where left matrix is a tridiagonal matrix.
Args | |
|---|---|
superdiag
|
A Tensor. Must be one of the following types: float64, float32, complex64, complex128.
Tensor of shape [..., 1, M], representing superdiagonals of
tri-diagonal matrices to the left of multiplication. Last element is ignored.
|
maindiag
|
A Tensor. Must have the same type as superdiag.
Tensor of shape [..., 1, M], representing main diagonals of tri-diagonal
matrices to the left of multiplication.
|
subdiag
|
A Tensor. Must have the same type as superdiag.
Tensor of shape [..., 1, M], representing subdiagonals of tri-diagonal
matrices to the left of multiplication. First element is ignored.
|
rhs
|
A Tensor. Must have the same type as superdiag.
Tensor of shape [..., M, N], representing MxN matrices to the right of
multiplication.
|
name
|
A name for the operation (optional). |
Returns | |
|---|---|
A Tensor. Has the same type as superdiag.
|