Component-wise multiplies a SparseTensor by a dense Tensor.
tf.raw_ops.SparseDenseCwiseMul( sp_indices, sp_values, sp_shape, dense, name=None )
The output locations corresponding to the implicitly zero elements in the sparse tensor will be zero (i.e., will not take up storage space), regardless of the contents of the dense tensor (even if it's +/-INF and that INF*0 == NaN).
Limitation: this Op only broadcasts the dense side to the sparse side, but not the other direction.
Args | |
---|---|
sp_indices | A Tensor of type int64 . 2-D. N x R matrix with the indices of non-empty values in a SparseTensor, possibly not in canonical ordering. |
sp_values | A Tensor . Must be one of the following types: float32 , float64 , int32 , uint8 , int16 , int8 , complex64 , int64 , qint8 , quint8 , qint32 , bfloat16 , uint16 , complex128 , half , uint32 , uint64 . 1-D. N non-empty values corresponding to sp_indices . |
sp_shape | A Tensor of type int64 . 1-D. Shape of the input SparseTensor. |
dense | A Tensor . Must have the same type as sp_values . R -D. The dense Tensor operand. |
name | A name for the operation (optional). |
Returns | |
---|---|
A Tensor . Has the same type as sp_values . |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/raw_ops/SparseDenseCwiseMul