tf.gather

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

根据维度和索引值来切片。

对params中维度为axis的值按照索引indices进行切片。

 

posted @ 2020-04-02 23:50  小千北同学超爱写代码  阅读(255)  评论(0)  编辑  收藏  举报