brainpy.math.unsorted_segment_sqrt_n

brainpy.math.unsorted_segment_sqrt_n(data, segment_ids, num_segments)

Computes the sum along segments of a tensor divided by the sqrt(N).

See https://tensorflow.google.cn/api_docs/python/tf/math/unsorted_segment_sqrt_n