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. CSR-разреженная матрица. |
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. |
© 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/SparseMatrixMatMul