jmaronas:

a=torch.zeros((10,10).cuda()


DataParallel does not meet my needs. There is simulator sampling I must do on CPU cores.
a.device.index does not give CPU index. I am looking for similar functionality but for CPU index