tensorflow::ops::SparseSegmentSqrtNWithNumSegments
#include <math_ops.h>
Computes the sum along sparse segments of a tensor divided by the sqrt of N.
Summary
N is the size of the segment being reduced.
Like SparseSegmentSqrtN
, but allows missing ids in segment_ids
. If an id is missing, the output
tensor at that position will be zeroed.
Read the section on segmentation for an explanation of segments.
Arguments:
- scope: A Scope object
- indices: A 1-D tensor. Has same rank as
segment_ids
. - segment_ids: A 1-D tensor. Values should be sorted and can be repeated.
- num_segments: Should equal the number of distinct segment IDs.
Returns:
-
Output
: Has same shape as data, except for dimension 0 which has sizek
, the number of segments.
Constructors and Destructors | |
---|---|
SparseSegmentSqrtNWithNumSegments(const ::tensorflow::Scope & scope, ::tensorflow::Input data, ::tensorflow::Input indices, ::tensorflow::Input segment_ids, ::tensorflow::Input num_segments) |
Public attributes | |
---|---|
operation | |
output |
Public functions | |
---|---|
node() const | ::tensorflow::Node * |
operator::tensorflow::Input() const | |
operator::tensorflow::Output() const |
Public attributes
operation
Operation operation
output
::tensorflow::Output output
Public functions
SparseSegmentSqrtNWithNumSegments
SparseSegmentSqrtNWithNumSegments( const ::tensorflow::Scope & scope, ::tensorflow::Input data, ::tensorflow::Input indices, ::tensorflow::Input segment_ids, ::tensorflow::Input num_segments )
node
::tensorflow::Node * node() const
operator::tensorflow::Input
operator::tensorflow::Input() const
operator::tensorflow::Output
operator::tensorflow::Output() const
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/cc/class/tensorflow/ops/sparse-segment-sqrt-n-with-num-segments