TensoFlow中的segment_sum函数
今天看TensorFlow自然语言处理这本书时,碰到了下面这个语句data = tf.constant([1,2,3,4,5,6,7,8,9,10],dtype=tf.float32)segment_ids = tf.constant([0,0,0,1,1,2,2,2,2,2],dtype=tf.int32)x_seg_sum = tf.segment_sum(data,segment_id...
原创
2020-01-08 17:20:55 ·
589 阅读 ·
0 评论