ATen
ATen copied to clipboard
support for sugment_sum
per tf definition https://www.tensorflow.org/api_docs/python/tf/segment_sum
Or if someone can point out how do we go about this one.
per tf definition https://www.tensorflow.org/api_docs/python/tf/segment_sum
Or if someone can point out how do we go about this one.