Adds two `SparseTensor` objects to produce another `SparseTensor`.
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.
Constants
String | OP_NAME | The name of this op, as known by TensorFlow core engine |
Public Methods
static <T extends TType> SparseAdd<T> | |
Output<TInt64> | |
Output<TInt64> |
sumShape()
|
Output<T> |
Inherited Methods
Constants
public static final String OP_NAME
The name of this op, as known by TensorFlow core engine
Public Methods
public static SparseAdd<T> create (Scope scope, Operand<TInt64> aIndices, Operand<T> aValues, Operand<TInt64> aShape, Operand<TInt64> bIndices, Operand<T> bValues, Operand<TInt64> bShape, Operand<? extends TNumber> thresh)
Factory method to create a class wrapping a new SparseAdd operation.
Parameters
scope | current scope |
---|---|
aIndices | 2-D. The `indices` of the first `SparseTensor`, size `[nnz, ndims]` Matrix. |
aValues | 1-D. The `values` of the first `SparseTensor`, size `[nnz]` Vector. |
aShape | 1-D. The `shape` of the first `SparseTensor`, size `[ndims]` Vector. |
bIndices | 2-D. The `indices` of the second `SparseTensor`, size `[nnz, ndims]` Matrix. |
bValues | 1-D. The `values` of the second `SparseTensor`, size `[nnz]` Vector. |
bShape | 1-D. The `shape` of the second `SparseTensor`, size `[ndims]` Vector. |
thresh | 0-D. The magnitude threshold that determines if an output value/index pair takes space. |
Returns
- a new instance of SparseAdd