标签： python tensorflow
假设我有一个a形状的张量[B,D]，并且我有一个列表I，其中包含形状为[B]的索引。现在，我想使用列表中的索引通过[M,D]将张量扩展为M > B的形状。请注意，索引属于范围[0,M]。具体来说，I是从张量a到另一个张量的映射，这些张量的维度为0。此功能与功能tf.gather()相反。 有人可以提出解决方案吗？ 谢谢
a
[B,D]
I
[B]
[M,D]
M > B
[0,M]
0
tf.gather()