tensorflow:: ops:: SparseSegmentMeanGradV2
#include <math_ops.h>
Computes gradients for SparseSegmentMean.
Summary
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:
- scope: A Scope object
- grad: gradient propagated to the SparseSegmentMean op.
- indices: indices passed to the corresponding SparseSegmentMean op.
- segment_ids: segment_ids passed to the corresponding SparseSegmentMean op.
- dense_output_dim0: dimension 0 of "data" passed to SparseSegmentMean op.
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
|
Public attributes
operation
Operation operation
output
::tensorflow::Output output
sorted_unique_indices
::tensorflow::Output sorted_unique_indices
Public functions
SparseSegmentMeanGradV2
SparseSegmentMeanGradV2( const ::tensorflow::Scope & scope, ::tensorflow::Input grad, ::tensorflow::Input indices, ::tensorflow::Input segment_ids, ::tensorflow::Input dense_output_dim0 )