tf.unsorted_segment_sqrt_n( data, segment_ids, num_segments, name=None )
Computes the sum along segments of a tensor divided by the sqrt(N).
Read the section on segmentation for an explanation of segments.
This operator is similar to the unsorted segment sum operator found here. Additionally to computing the sum over segments, it divides the results by sqrt(N).
\(output_i = 1/sqrt(N_i) \sum data_j\) where the sum is over
segment_ids[j] == i with \N_i\ being the number of occurrences
of id \i\.
If there is no entry for a given segment ID
i, it outputs 0.
Note that this op only supports floating point and complex dtypes, due to tf.sqrt only supporting these types.
segment_ids: A 1-D tensor whose rank is equal to the rank of
output: Has same shape as data, except for dimension 0 which