变量名.type()转换tensor类型的时候注意!!!
torch.cuda.LongTensor:加载到GPU
torch.LongTensor:加载到CPU
官网其他转换类型
model.cuda()
input_iter = Variable(torch.tensor(input_iter))
input_iter = input_iter.type(torch.cuda.LongTensor)
output = model(input_iter)
变量名.type()转换tensor类型的时候注意!!!
torch.cuda.LongTensor:加载到GPU
torch.LongTensor:加载到CPU
官网其他转换类型
model.cuda()
input_iter = Variable(torch.tensor(input_iter))
input_iter = input_iter.type(torch.cuda.LongTensor)
output = model(input_iter)