根据Tensorflow中的其他特定张量重新调整张量

时间:2018-05-21 11:37:28

标签: python tensorflow

鉴于某些张量ab其中b属于uintmax(b) < len(a),我希望得到一个张量c

c = a[b]

根据a给出的顺序,这基本上会重新调整张量b

1 个答案:

答案 0 :(得分:0)

关于这个问题 - 根据@jdehesa的评论:

c = tf.gather(a, b)

这是完全需要的。谢谢@jdehesa!