RuntimeError: mat1 dim 1 must match mat2 dim 0
这个报错原因是连接层维度不匹配,全连接神经网络中 dim一般为2
第0维为样本数 第1维为特征数 当时因为报这个错是因为做一个线性预测,而特征数只有1个 一次batch-size的训练tensor的shape为[batch-size],这维度会报错,需要view成[batch-size,1],其中1是特征数
RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor)
这个报错原因是 我们在进行正向传播将样本送进去的时候 已经将样本数据迁移到cuda上 还得将该样本tensor类型转为double
即正向传播接收的tensor类型为double