brainpy.math.unsorted_segment_sum#

brainpy.math.unsorted_segment_sum(data, segment_ids, num_segments)[source]#

Computes the sum along segments of a tensor.

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