Hi @tom.  That would work if I had values_batch * 2, but actually the number I have is a float, as in: values_batch * 0.5. So if i convert the tensor to torch.uint8 , my floating point would get rounded to the nearest integer. I’ll try to figure some other way around to do the indexing. Thanks for your help!