#include <math_ops.h>
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".
Args:
Returns:
| Constructors and Destructors | |
|---|---|
SparseSegmentMeanGradV2(const ::tensorflow::Scope & scope, ::tensorflow::Input grad, ::tensorflow::Input indices, ::tensorflow::Input segment_ids, ::tensorflow::Input dense_output_dim0) |
| Public attributes | |
|---|---|
operation | |
output | |
sorted_unique_indices | |
Operation operation
::tensorflow::Output output
::tensorflow::Output sorted_unique_indices
SparseSegmentMeanGradV2( const ::tensorflow::Scope & scope, ::tensorflow::Input grad, ::tensorflow::Input indices, ::tensorflow::Input segment_ids, ::tensorflow::Input dense_output_dim0 )
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/cc/class/tensorflow/ops/sparse-segment-mean-grad-v2