brainpy.math.unsorted_segment_max#

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

Computes the maximum along segments of a tensor.

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