数据cpu和cuda的问题

Tensor 有三种属性： dtype， device，和layout

问题： 数显cuda和cpu混乱，无法互相计算
解决：

device = torch.device('cuda', if torch.cuda.is_device()else 'cpu' )
x = torch.rand(1,1,3,4)
x = x.data.to(device)