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