tf.raw_ops.SparseMatrixMul

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

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.raw_ops.SparseMatrixMul

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.