如何使用 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])