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> |
Inherited Methods
Constants
public static final String OP_NAME
The name of this op, as known by TensorFlow core engine
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