tf.gather(params,indices,axis=0 )
从params的axis维根据indices的参数值获取切片

如上图所示,params一共6个维度,indices为[2,1,3,4]被取了出来。

本文介绍TensorFlow中tf.gather函数的使用方法,该函数能够从指定的轴上根据索引值提取切片。通过示例展示如何利用索引[2,1,3,4]从六维参数中选取特定元素。
tf.gather(params,indices,axis=0 )
从params的axis维根据indices的参数值获取切片

如上图所示,params一共6个维度,indices为[2,1,3,4]被取了出来。

1668

被折叠的 条评论
为什么被折叠?