Как удалить элементы из 3D-тензора с помощью PyTorch?

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:]
t.shape
torch.Size([4, 161, 324])