python - 如何使用 PyTorch 从 3D 张量中删除元素？

Question

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

标签： pythonpytorchtensor

Answer 1

你可以使用简单的切片，

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