PyTorch Forums
Legitimate way to move module or tensor to GPU?
vision
SimonW
(Simon Wang)
September 15, 2018, 2:45am
2
No
.to is preferred because you can define a device object and use it everywhere.
1 Like
show post in topic