tfep.nn.graph.unsorted_segment_sum

tfep.nn.graph.unsorted_segment_sum(data, segment_ids, n_segments)[source]

Replicates TensorFlow’s tf.math.unsorted_segment_sum in PyTorch.