blingkeyholic的博客

原创 tf.gather_nd和tf.gather的区别与联系

tf.gather_nd和tf.gather的区别与联系tf.gather_nd和tf.gather都是选择其中需要的部分组成一个新的矩阵。其中tf.gather(params,indices,axis=0.name=None)是选择相同维度的元素组成新的矩阵，params是待处理tensor,indices是一个1-D tensor表示对应轴上的索引，axis是指定维度，name是操作名字。...