NOTES
API Reference
segment softmax function.
data – The source tensor.
segment_ids – The indices of elements for applying the softmax.
num_segments – The number of segments.
softmax score.
tensor