tf.raw_ops.TridiagonalMatMul
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 .
|
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.
Last updated 2023-10-06 UTC.
[null,null,["Last updated 2023-10-06 UTC."],[],[]]