这个是在最近看代码中看到的问题, 有时候我们在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)
