【无标题】

将数据放到GPU上，

    device = torch.device('cuda')
    alpha=torch.tensor(alpha)
    alpha = alpha.to(device)
    正确

若直接alpha = alpha.cuda()  #'list'/'numpy' object has no attribute 'cuda'
或者直接alpha = alpha.to(device)，报错：alpha没有device属性

A.is_cuda判断A变量‘A’是否在GPU上，返回 True或者False