xxxxxxxxxx
#Tensorflow is a popular, free, open-source software library for machine learning and AI. But It specializes on DNN training and inference.
import tensorflow as tf #import tensorflow in python
print(tf.__version__) #print tensorflow version
xxxxxxxxxx
c = tf.constant([[1,2,3,4], [-1,-2,-3,-4], [5,6,7,8]])
tf.math.segment_sum(c, tf.constant([0, 0, 1]))
# ==> [[0 0 0 0]
# [5 6 7 8]]