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 — ложь, возвращает:
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 | A Tensor типа variant. CSRSparseMatrix. |
b | A 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 | Имя операции (необязательно). |
| Возвращаемые значения | |
|---|---|
A 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/versions/r2.9/api_docs/python/tf/raw_ops/SparseMatrixMatMul