码迷,mamicode.com
首页 > 其他好文 > 详细

tensorflow tensor 索引

时间:2019-01-14 23:07:21      阅读:212      评论:0      收藏:0      [点我收藏+]

标签:int   long   pytho   eval   use   none   tps   useful   name   

 

 

 

 

问题:

self.q_eval4next: (100,2)

ix=【0,1,0,1---0,1】(100,1)
我想取q_eval4next[:,idx]

 


 

 1         #use_doubleQ  切片用!!!!
 2         self.range_index = tf.placeholder(tf.int32,[None,],name=range_index)
 3 
 4         if self.use_doubleQ:
 5 
 6             f = tf.map_fn(lambda x: x, self.range_index) # or perhaps something more useful than identity
 7             ix = tf.to_int32(tf.expand_dims(tf.argmax(self.q_eval4next,axis=1),-1))
 8             tmp=tf.to_int32(tf.expand_dims(f,-1))
 9             index_a = tf.concat([tmp,ix,],axis=1)
10             maxq =  tf.gather_nd(self.q_next,index_a)

 https://www.programcreek.com/python/example/90420/tensorflow.map_fn

https://stackoverflow.com/questions/34987509/tensorflow-max-of-a-tensor-along-an-axis

https://zhuanlan.zhihu.com/p/39295071

https://zhuanlan.zhihu.com/p/45673869

tensorflow tensor 索引

标签:int   long   pytho   eval   use   none   tps   useful   name   

原文地址:https://www.cnblogs.com/zle1992/p/10269187.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!