tf.gather( )的用法

tf.gather(params,indices,axis=0 )

从params的axis维根据indices的参数值获取切片

tf.gather( )的用法_第1张图片

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

tf.gather( )的用法_第2张图片

 

你可能感兴趣的:(函数用法)