如何使用 PyTorch 從 3D 張量中刪除元素？

Question

我有一個形狀torch.Size([4, 161, 325])的張量。 如何刪除沿 dim=2 的第一個元素，以便生成的張量具有torch.Size([4, 161, 324])的形狀？

Answer 1

你可以使用簡單的切片，

>>>a = torch.randn(4, 161, 325)
>>>b = a[:, :, 1:]
>>>b.shape
torch.Size([4, 161, 324])

Answer 2

切片

t = torch.rand(4,161,325)

t = t[..., 1:]            # or t = t[Ellipsis, 1:] Here, Ellipsis indicate rest of dims
t.shape
torch.Size([4, 161, 324])