tf.math.cumsum