I have multiple GPUs and im using dataparallel.
Every time i print my tensor in my model it shows that it is running on one of the GPUs (e.g. GPU 0 or GPU 1, or GPU 2)
is there a way to get the gpu index that the tensor is using  at each time(i.e. 0, or 1 or 2)?