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> | |
Output<T> |
output()
|
Output<U> |
Inherited Methods
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