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.
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.
|