复制Pytorch中某些索引的张量元素

Question

所需的操作在精神上与torch.tensor.index_copy相似，但有些不同。

最好用一个例子来解释。

张量a具有我们将复制的原始值：

[10，20，30]

张量b具有a ：

[0、1、0、1、2、1]

张量c的长度与b，包含<索引值<代码> a ：

[10、20、10、20、30、20]

是从a c的好方法>和b在pytorch中，不使用循环？

Answer 1

您是否尝试过仅通过A索引？

In [1]: import torch
  
In [2]: a = torch.tensor([20,30,40])

In [3]: b = torch.tensor([0,1,2,1,1,2,0,0,1,2])

In [4]: a[b]
Out[4]: tensor([20, 30, 40, 30, 30, 40, 20, 20, 30, 40])