Any equivalent to tf.segment_max

[be7f984b2f2e66ba7969 (杜明軒)]

Hi,

in tensorflow, we have tf.segment_max (https://www.tensorflow.org/api_docs/python/tf/segment_max)

but in pytorch, what is the equivalent way to do this?

thanks