tf.raw_ops.TridiagonalMatMul

Вычисление произведения с тридиагональной матрицей.

Псевдонимы для миграции

Дополнительную информацию см. в руководстве по миграции.

tf.raw_ops.TridiagonalMatMul(
    superdiag, maindiag, subdiag, rhs, name=None
)

Вычисляет произведение двух матриц, где левая матрица является тридиагональной.

Аргументы
`superdiag`	A `Tensor`. Должен быть одним из следующих типов: `float64`, `float32`, `complex64`, `complex128`. Tensor с формой `[..., 1, M]`, представляющий наддиагональные элементы тридиагональных матриц слева от умножения. Последний элемент игнорируется.
`maindiag`	A `Tensor`. Должен иметь тот же тип, что и `superdiag`. Tensor с формой `[..., 1, M]`, представляющий главные диагональные элементы тридиагональных матриц слева от умножения.
`subdiag`	A `Tensor`. Должен иметь тот же тип, что и `superdiag`. Tensor с формой `[..., 1, M]`, представляющий поддиагональные элементы тридиагональных матриц слева от умножения. Первый элемент игнорируется.
`rhs`	A `Tensor`. Должен иметь тот же тип, что и `superdiag`. Tensor с формой `[..., 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.4/api_docs/python/tf/raw_ops/TridiagonalMatMul