tf.gather tf.gather_nd 和 tf.batch_gather 使用方法
在计算机视觉的项目中,会遇到根据索引获取数组特定下标元素的任务。常用的函数有tf.gather , tf.gather_nd 和 tf.batch_gather。1.tf.gather(params,indices,validate_indices=None,name=None,axis=0)主要参数:params:被索引的张量 indic...
原创
2019-01-19 15:09:33 ·
13257 阅读 ·
5 评论