Spec-Zone.ru › TensorFlow 2.3

tf.raw_ops.TridiagonalMatMul

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

Просмотр псевдонимов

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

См. Руководство по миграции для получения более подробной информации.

tf.compat.v1.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

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API