Spec-Zone.ru › TensorFlow

tf.raw_ops.SparseMatrixMatMul

Перемножает разреженную матрицу на плотную матрицу.

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

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

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

tf.compat.v1.raw_ops.SparseMatrixMatMul

tf.raw_ops.SparseMatrixMatMul(
    a,
    b,
    transpose_a=False,
    transpose_b=False,
    adjoint_a=False,
    adjoint_b=False,
    transpose_output=False,
    conjugate_output=False,
    name=None
)

Возвращает плотную матрицу. Для входных данных A и B, где A — CSR, а B — плотная матрица, эта операция возвращает плотную матрицу C;

Если transpose_output имеет значение false, возвращает:

C = A . B

Если transpose_output имеет значение true, возвращает:

C = transpose(A . B) = transpose(B) . transpose(A)

где транспонирование выполняется по двум самым внутренним измерениям (матрицы).

Если conjugate_output имеет значение true, возвращает:

C = conjugate(A . B) = conjugate(A) . conjugate(B)

Если conjugate_output и transpose_output имеют значение true, возвращает:

C = conjugate(transpose(A . B)) = conjugate(transpose(B)) .
                                  conjugate(transpose(A))
Аргументы
a Tensor типа variant. Разреженная матрица CSR.
b Tensor. Плотный тензор.
transpose_a Необязательный bool. По умолчанию — False. Указывает, следует ли транспонировать a.
transpose_b Необязательный bool. По умолчанию — False. Указывает, следует ли транспонировать b.
adjoint_a Необязательный bool. По умолчанию — False. Указывает, следует ли выполнить сопряжённое транспонирование a.
adjoint_b Необязательный bool. По умолчанию — False. Указывает, следует ли выполнить сопряжённое транспонирование b.
transpose_output Необязательный bool. По умолчанию — False. Транспонирует произведение a и b.
conjugate_output Необязательный bool. По умолчанию — False. Сопрягает произведение a и b.
name Имя операции (необязательно).
Возвращает
Tensor. Имеет тот же тип, что и b.

© 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/SparseMatrixMatMul

Spec-Zone.ru

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