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