火炬补充索引操作

Question

我有一个张量说t = [1、2、3、4、5、6、7、8、9]和我想删除的索引列表，例如d = [0 ，2，5]。

做t [d]给我带有我要删除的元素的子镜。

我如何使用d来获取所有元素 d索引的子镜头。必须有t [〜d]之类的东西，对吗？

像Numpy的 numpy.delete.delete.delete 。

Answer 1

delete使用逻辑，例如

mask= np.ones(t.shape, dtype=bool)
mask[d] = False
res = t[mask]

mask是我们要保留的所有索引的bool。这是简单的1D版本。

In [3]: t=np.arange(1,10); d= np.array([0,2,5])

In [4]: mask = np.ones(t.shape, bool)
In [5]: mask[d] = False
In [6]: mask
Out[6]: array([False,  True, False,  True,  True, False,  True,  True,  True])

In [7]: t[mask]
Out[7]: array([2, 4, 5, 7, 8, 9])