tf.raw_ops.TridiagonalMatMul
Вычисление произведения с трёхдиагональной матрицей.
tf.raw_ops.TridiagonalMatMul(
superdiag, maindiag, subdiag, rhs, name=None
)
Вычисляет произведение двух матриц, где левая матрица является трёхдиагональной матрицей.
| Аргументы | |
|---|---|
superdiag | A Tensor. Должно быть одним из следующих типов: float64, float32, complex64, complex128. Тензор формы [..., 1, M], представляющий наддиагональные элементы трёхдиагональных матриц слева от умножения. Последний элемент игнорируется. |
maindiag | A Tensor. Должен иметь тот же тип, что и superdiag. Тензор формы [..., 1, M], представляющий главные диагональные элементы трёхдиагональных матриц слева от умножения. |
subdiag | A Tensor. Должен иметь тот же тип, что и superdiag. Тензор формы [..., 1, M], представляющий поддиагональные элементы трёхдиагональных матриц слева от умножения. Первый элемент игнорируется. |
rhs | A Tensor. Должен иметь тот же тип, что и superdiag. Тензор формы [..., M, N], представляющий матрицы MxN справа от умножения. |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
A Tensor. Имеет тот же тип, что и superdiag. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/raw_ops/TridiagonalMatMul