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`.

© 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/versions/r2.9/api_docs/python/tf/raw_ops/TridiagonalMatMul