Class SparseSegmentSumGrad<T extends TNumber, U extends TNumber>

java.lang.Object
org.tensorflow.op.RawOp
org.tensorflow.op.sparse.SparseSegmentSumGrad<T,U>
All Implemented Interfaces:
Op

@Operator(group="sparse") public final class SparseSegmentSumGrad<T extends TNumber, U extends TNumber> extends RawOp
Computes gradients for SparseSegmentSum. 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".
  • Field Details

  • Constructor Details

    • SparseSegmentSumGrad

      public SparseSegmentSumGrad(Operation operation)
  • Method Details

    • create

      @Endpoint(describeByClass=true) public static <T extends TNumber, U extends TNumber> SparseSegmentSumGrad<T,U> create(Scope scope, Operand<T> grad, Operand<U> indices, Operand<? extends TNumber> segmentIds, Operand<TInt32> denseOutputDim0)
      Factory method to create a class wrapping a new SparseSegmentSumGradV2 operation.
      Type Parameters:
      T - data type for SparseSegmentSumGradV2 output and operands
      U - data type for SparseSegmentSumGradV2 output and operands
      Parameters:
      scope - current scope
      grad - gradient propagated to the SparseSegmentSum op.
      indices - indices passed to the corresponding SparseSegmentSum op.
      segmentIds - segment_ids passed to the corresponding SparseSegmentSum op.
      denseOutputDim0 - dimension 0 of "data" passed to SparseSegmentSum op.
      Returns:
      a new instance of SparseSegmentSumGrad
    • output

      public Output<T> output()
      Gets output.
      Returns:
      output.
    • sortedUniqueIndices

      public Output<U> sortedUniqueIndices()
      Gets sortedUniqueIndices.
      Returns:
      sortedUniqueIndices.