SparseAddGrad

public final class SparseAddGrad

The gradient operator for the SparseAdd op.

The SparseAdd op calculates A + B, where A, B, and the sum are all represented as `SparseTensor` objects. This op takes in the upstream gradient w.r.t. non-empty values of the sum, and outputs the gradients w.r.t. the non-empty values of A and B.

Constants

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

Public Methods

Output<T>
aValGrad()
1-D with shape `[nnz(A)]`.
Output<T>
bValGrad()
1-D with shape `[nnz(B)]`.
static <T extends TType> SparseAddGrad<T>
create(Scope scope, Operand<T> backpropValGrad, Operand<TInt64> aIndices, Operand<TInt64> bIndices, Operand<TInt64> sumIndices)
Factory method to create a class wrapping a new SparseAddGrad operation.

Inherited Methods

Constants

public static final String OP_NAME

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

Constant Value: "SparseAddGrad"

Public Methods

public Output<T> aValGrad ()

1-D with shape `[nnz(A)]`. The gradient with respect to the non-empty values of A.

public Output<T> bValGrad ()

1-D with shape `[nnz(B)]`. The gradient with respect to the non-empty values of B.

public static SparseAddGrad<T> create (Scope scope, Operand<T> backpropValGrad, Operand<TInt64> aIndices, Operand<TInt64> bIndices, Operand<TInt64> sumIndices)

Factory method to create a class wrapping a new SparseAddGrad operation.

Parameters
scope current scope
backpropValGrad 1-D with shape `[nnz(sum)]`. The gradient with respect to the non-empty values of the sum.
aIndices 2-D. The `indices` of the `SparseTensor` A, size `[nnz(A), ndims]`.
bIndices 2-D. The `indices` of the `SparseTensor` B, size `[nnz(B), ndims]`.
sumIndices 2-D. The `indices` of the sum `SparseTensor`, size `[nnz(sum), ndims]`.
Returns
  • a new instance of SparseAddGrad