2020-01-15 将数据从CPU赋值到GPU变量上

这个是在最近看代码中看到的问题, 有时候我们在GPU上已经定义好了变量,但是需要的数据却是在CPU上.

那么这个时候我们可能会用 resize_as_ + copy 的方式来解决这个问题.

  for i, data in enumerate(train_loader):
        real_cpu, _ = data
        input = torch.FloatTensor(64, 3, 64, 64)
        input.resize_as_(real_cpu).copy_(real_cpu)  

在这里可以这么用,这段代码出自D2GAN, 感兴趣的朋友可以看看.

最后,祝大家新年快乐!!!!

你可能感兴趣的:(2020-01-15 将数据从CPU赋值到GPU变量上)