Is it possible to implement segment_min function from TensorFlow in PyTorch? Here is details of this method: tf.math.segment_min | TensorFlow Core v2.4.1
Based on the docs I think the operation would correspond to scatter_min_
from PyTorch Scatter.
There is also an open feature request here.