def METHOD_NAME(self, msg):
    """This method reduce message by mean. 
    Args:
        feature (paddle.Tensor): feature with first dim as num_edges.
    Returns:
        Returns a paddle.Tensor with the first dim the same as the largest segment_id.
    """
    return math.segment_mean(msg, self._segment_ids)