tf.raw_ops.SparseMatrixMul
Элементное умножение разреженной матрицы на плотное тензорное значение.
tf.raw_ops.SparseMatrixMul(
a, b, name=None
)
Возвращает разреженную матрицу.
Плотный тензор b может быть скалярным; в противном случае a должен быть тензором ранга 3 SparseMatrix; в этом случае b должен иметь форму [batch_size, 1, 1] и операция умножения выполняется с трансляцией.
Примечание: даже если b равен нулю, структура разреженности выходного значения не изменяется.
| Аргументы | |
|---|---|
a | A Tensor типа variant. CSRSparseMatrix. |
b | A Tensor. Плотный тензор. |
name | Имя операции (необязательно). |
| Возвращаемые значения | |
|---|---|
A Tensor типа variant. |
© 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/SparseMatrixMul