把數據遷移到GPU: .cuda()方法和.to(device)方法

把數據從CPU遷移到GPU時，可以用.cuda()方法，也可以用.to(device)方法。示例如下。 .cuda()方法 import torch import time t = tim ...