Similarly I need the equivalent of tf.unsorted_segment_sum 16. Such segment reduction ops were implemented in Caffe2 5, which seem to be getting merged into PyTorch: https://github.com/pytorch/pytorch/blob/master/caffe2/operators/segment_reduction_op.cc 84
Hopefully they’ll be available in PyTorch 1.0…