标签: python tensorflow
鉴于某些张量a和b其中b属于uint和max(b) < len(a),我希望得到一个张量c :
a
b
uint
max(b) < len(a)
c
c = a[b]
根据a给出的顺序,这基本上会重新调整张量b。
答案 0 :(得分:0)
关于这个问题 - 根据@jdehesa的评论:
c = tf.gather(a, b)
这是完全需要的。谢谢@jdehesa!