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 | A Tensor типа variant. A 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/api_docs/python/tf/raw_ops/SparseMatrixMatMul