SparseAdd

Stay organized with collections Save and categorize content based on your preferences.
public final class SparseAdd

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>
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.
Output<TInt64>
Output<TInt64>
Output<T>

Inherited Methods

Constants

public static final String OP_NAME

The name of this op, as known by TensorFlow core engine

Constant Value: "SparseAdd"

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

public Output<TInt64> sumIndices ()

public Output<TInt64> sumShape ()

public Output<T> sumValues ()