I have a torch tensor of size torch.Size([10, 1, 224, 224]), holding 10 images of size 224x224. I would like to extract single images but keep the tensor size as torch.Size([1, 1, 224, 224]). How can I achieve this?