tf.raw_ops.SparseMatrixMul

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

View aliases

Compat aliases for migration

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.

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

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.

Last updated 2022-10-27 UTC.

tf.raw_ops.SparseMatrixMul

View aliases

Args

Returns