Hacker News new | ask | show | jobs
by sva_ 1202 days ago
Or better yet, define a device = 'cpu', and use tensor.to(device).