To have all the tensors on the same device
Thank you for your submission. ^_^
· Sign up or log in to comment