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