brainpy.math.unsorted_segment_prod#

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

Computes the product along segments of a tensor.

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