我直接把float类型的数据传入给深度学习网络,出现了问题
报错为“Expected object of scalar type Float but got scalar type Double for argument #2 ‘mat1’ in call to _th_addmm”
解决方式,在输入数据的后面增加了type(torch.FloatTensor)方法
x = clean_data[:, 1:22].type(torch.FloatTensor)
不知道为什么,我之前已经转化了一次张量,先这样吧
y = torch.cat((y0, y1)).type(torch.LongTensor)
y是标签,也要进行tensor的转化,只不过是用longtensor