def test_UnsortedSegmentSum(self):
t = tf.unsorted_segment_sum(self.random(4, 2, 3), np.array([0, 2, 2, 1]), 3)
self.check(t)
微信
公众号
扫码关注公众号