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