tf.raw_ops.TridiagonalMatMul

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

Совместимые псевдонимы для миграции

Подробнее см. в руководстве по миграции.

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

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

Аргументы
`superdiag`	Тензор `Tensor`. Должен иметь один из следующих типов: `float64`, `float32`, `complex64`, `complex128`. Тензор формы `[..., 1, M]`, представляющий наддиагонали трёхдиагональных матриц слева от знака умножения. Последний элемент игнорируется.
`maindiag`	Тензор `Tensor`. Должен иметь тот же тип, что и `superdiag`. Тензор формы `[..., 1, M]`, представляющий главные диагонали трёхдиагональных матриц слева от знака умножения.
`subdiag`	Тензор `Tensor`. Должен иметь тот же тип, что и `superdiag`. Тензор формы `[..., 1, M]`, представляющий поддиагонали трёхдиагональных матриц слева от знака умножения. Первый элемент игнорируется.
`rhs`	Тензор `Tensor`. Должен иметь тот же тип, что и `superdiag`. Тензор формы `[..., M, N]`, представляющий матрицы MxN справа от знака умножения.
`name`	Имя операции (необязательно).

Возвращаемое значение
Тензор `Tensor`. Имеет тот же тип, что и `superdiag`.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/raw_ops/TridiagonalMatMul