tf.raw_ops.SparseAdd

Adds two SparseTensor objects to produce another SparseTensor.

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.raw_ops.SparseAdd

tf.raw_ops.SparseAdd(
    a_indices, a_values, a_shape, b_indices, b_values, b_shape, thresh, name=None
)

The input SparseTensor objects' indices are assumed ordered in standard lexicographic order. If this is not the case, before this step run SparseReorder to restore index ordering.

By default, if two values sum to zero at some index, the output SparseTensor would still include that particular location in its index, storing a zero in the corresponding value slot. To override this, callers can specify thresh, indicating that if the sum has a magnitude strictly smaller than thresh, its corresponding value and index would then not be included. In particular, thresh == 0 (default) means everything is kept and actual thresholding happens only for a positive value.

In the following shapes, nnz is the count after taking thresh into account.

Args
a_indices	A Tensor of type int64. 2-D. The indices of the first SparseTensor, size [nnz, ndims] Matrix.
a_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. The values of the first SparseTensor, size [nnz] Vector.
a_shape	A Tensor of type int64. 1-D. The shape of the first SparseTensor, size [ndims] Vector.
b_indices	A Tensor of type int64. 2-D. The indices of the second SparseTensor, size [nnz, ndims] Matrix.
b_values	A Tensor. Must have the same type as a_values. 1-D. The values of the second SparseTensor, size [nnz] Vector.
b_shape	A Tensor of type int64. 1-D. The shape of the second SparseTensor, size [ndims] Vector.
thresh	A Tensor. Must be one of the following types: float32, float64, int32, uint8, int16, int8, int64, bfloat16, uint16, half, uint32, uint64. 0-D. The magnitude threshold that determines if an output value/index pair takes space.
name	A name for the operation (optional).

Returns
A tuple of Tensor objects (sum_indices, sum_values, sum_shape).
sum_indices	A Tensor of type int64.
sum_values	A Tensor. Has the same type as a_values.
sum_shape	A Tensor of type int64.