如何通过在pytorch中使用除给定索引之外的所有索引来索引张量？答案

【问题标题】：How to index tensor by using all but given indices in pytorch?如何通过在pytorch中使用除给定索引之外的所有索引来索引张量？
【发布时间】：2022-01-27 04:35:00
【问题描述】：

假设我有一个张量a

>>> a
tensor([[2, 8],
        [3, 0],
        [4, 2],
        [2, 2],
        [6, 8]])

我还有一个张量 idx，其中包含我不想选择的索引

>>> idx
tensor([0, 3, 4])

我应该如何继续并选择除索引存在于张量 idx 中的行之外的所有行？对于上面的张量a和idx，选择后我应该得到输出张量b为

>>> b
tensor([[3, 0],
        [4, 2]])

【问题讨论】：

【解决方案1】：

a[[i for i in range(len(a)) if i not in idx]]

【讨论】：