index a list of tensors
I have a list object named &quot; imgs &quot; of tensors (50 images).
I have an array of indices (indi) of length 29.
how do I index the list of tensors with the array of indices?
when I do the following I get:
imgs[indi] 

TypeError: only integer scalar arrays can be converted to a scalar index

Thanks

Assuming these are normal python lists then you can use a list comprehension
result = [imgs[i] for i in indi]

which will give a list of tensors.
If you further want to make this a single tensor containing the images you can use torch.stack
result = torch.stack([imgs[i] for i in indi], dim=0)

