tf.raw_ops.SparseMatrixMul
Stay organized with collections
Save and categorize content based on your preferences.
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.
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.
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 2024-04-26 UTC.
[null,null,["Last updated 2024-04-26 UTC."],[],[]]
