Delete row at index

1

I have a 4-dimensional tensor and I want to select all the elements from that tensor except for one row along a dimension. How can I do that?
Here is the code snippet I have

features = Variable(torch.rand(2, 128, 19, 128))
selected = features[:, :, ~2, :] # I want to select all rows except the 2nd row along 3rd dimension.

2

You could use a double for loop and iterate through the tensor removing the second rows. Im sure there are better ways to do this but seeing how your tensor isn’t that big it should take that long.