org.tensorflow.op.core

Class SparseAddGrad<T>

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.core.SparseAddGrad<T>

Type Parameters:

T - data type for aValGrad() output

All Implemented Interfaces:

Op

@Operator
public final class SparseAddGrad<T>
extends PrimitiveOp

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.

Field Summary

Fields inherited from class org.tensorflow.op.PrimitiveOp

operation

Method Summary

Modifier and Type	Method and Description
Output<T>	aValGrad() 1-D with shape `[nnz(A)]`.
Output<T>	bValGrad() 1-D with shape `[nnz(B)]`.
static <T> SparseAddGrad<T>	create(Scope scope, Operand<T> backpropValGrad, Operand<Long> aIndices, Operand<Long> bIndices, Operand<Long> sumIndices) Factory method to create a class to wrap a new SparseAddGrad operation to the graph.

Methods inherited from class org.tensorflow.op.PrimitiveOp

equals, hashCode, toString

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Method Detail

create

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

Factory method to create a class to wrap a new SparseAddGrad operation to the graph.

Parameters:

scope - current graph 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

aValGrad

public Output<T> aValGrad()

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

bValGrad

public Output<T> bValGrad()

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