标签： tensorflow
假设我有一个浮点数的二维张量A和一个一维张量B的整数。 B中的数字表示A行的索引。如何在张量流图中有效地执行此索引的查找？
答案 0 :(得分：0)
您是否已尝试tf.gather()？
另请参阅此问题：TensorFlow: using a tensor to index another tensor