def test_SegmentSum(self): t = tf.segment_sum(self.random(4, 2, 3), np.array([0, 1, 1, 2])) self.check(t)