brainpy.math.unsorted_segment_mean#

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

Computes the average along segments of a tensor.

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