SparseSegmentMeanGradV2

public final class SparseSegmentMeanGradV2

Computes gradients for SparseSegmentMean.

Returns tensor "output" with same shape as grad, except for dimension 0 whose value is the number of unique indexes in "indices". Also returns vector "sorted_unique_indices" containing the corresponding indexes from "indices".

Public Methods

static <T extends Number, U extends Number, V extends Number> SparseSegmentMeanGradV2<T, U>
create(Scope scope, Operand<T> grad, Operand<U> indices, Operand<V> segmentIds, Operand<Integer> denseOutputDim0)
Factory method to create a class wrapping a new SparseSegmentMeanGradV2 operation.
Output<T>
output()
Output<U>
sortedUniqueIndices()

Inherited Methods

org.tensorflow.op.PrimitiveOp
final boolean
equals(Object obj)
final int
hashCode()
Operation
op()
Returns the underlying Operation
final String
toString()
boolean
equals(Object arg0)
final Class<?>
getClass()
int
hashCode()
final void
notify()
final void
notifyAll()
String
toString()
final void
wait(long arg0, int arg1)
final void
wait(long arg0)
final void
wait()

Public Methods

public static SparseSegmentMeanGradV2<T, U> create (Scope scope, Operand<T> grad, Operand<U> indices, Operand<V> segmentIds, Operand<Integer> denseOutputDim0)

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

Parameters
scope current scope
grad gradient propagated to the SparseSegmentMean op.
indices indices passed to the corresponding SparseSegmentMean op.
segmentIds segment_ids passed to the corresponding SparseSegmentMean op.
denseOutputDim0 dimension 0 of "data" passed to SparseSegmentMean op.
Returns
  • a new instance of SparseSegmentMeanGradV2

public Output<T> output ()

public Output<U> sortedUniqueIndices ()