W3cubDocs

/TensorFlow 2.4

tf.raw_ops.SparseMatrixMul

Element-wise multiplication of a sparse matrix with a dense tensor.

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.

© 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.4/api_docs/python/tf/raw_ops/SparseMatrixMul