保险起见 开头加上这句
torch.set_default_tensor_type(torch.DoubleTensor)
data[0] = data[0].to(torch.double)
data[1] = data[1].to(torch.double)
同时转换为double类型的数据
保险起见 开头加上这句
torch.set_default_tensor_type(torch.DoubleTensor)
data[0] = data[0].to(torch.double)
data[1] = data[1].to(torch.double)
同时转换为double类型的数据