tf.raw_ops.SparseMatrixMul
Element-wise multiplication of a sparse matrix with a dense tensor.
tf.raw_ops.SparseMatrixMul(
a, b, name=None
)
Returns a sparse matrix.
The dense tensor b may be either a scalar; otherwise a must be a rank-3 SparseMatrix; in this case b must be shaped [batch_size, 1, 1] and the multiply operation broadcasts.
Note: even if b is zero, the sparsity structure of the output does not change.
| Args | |
|---|---|
a |
A Tensor of type variant. A CSRSparseMatrix. |
b |
A Tensor. A dense tensor. |
name |
A name for the operation (optional). |
| Returns | |
|---|---|
A Tensor of type 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