RuntimeError: Expected object of scalar type Float but got scalar type Double for argument #4 'mat1'

RuntimeError: Expected object of scalar type Float but got scalar type Double for argument #4 ‘mat1’

今天，使用pytorch做一个小实验，结果报了上面的这个错误。查了很多资料都没有用。最后一点点调试解决了。
我本来打算实验kaggle上csv格式的mnist数据集来做一个简单的网络。因为直接使用pytorch里自带的还要下载，速度太慢。但是，却遇到了这个错误。

报错原图：

解决办法

先直接说解决办法。
这个错提示的是本该是float的输入，结果却是double。问题就在这个地方。我查看了数据的类型，确实是float的。只不过是np.float64。我把它改成np.float32就行了。

部分代码

import pandas as pd
import numpy as np

from models import  simpleNet
from torch.autograd import Variable
import torch

train = pd.read_csv('data/train.csv')

# 由于全部的mnist训练起来有点慢，所以我们只取一部分
X = train[:1000]
y = X['label']

X = X.drop(labels=['label'], axis=1)
X = X / 255.0

# 将x,y转为tensor
X, y = np.array(X), np.array(y)
print(type(X[0][0]))       # 结果是float64
X = X.astype(np.float32)   # 就是这一句修改数据类型，否则就会出错

X, y = torch.from_numpy(X), torch.from_numpy(y)

net = simpleNet(28*28, 100, 100, 10)

inputs = torch.randn(1, 784)
a = X
a = a[:, np.newaxis]

print(a[0].shape)    #[1, 784]
print(inputs.shape)  # [1, 784]
print(a[0])    
print(net(inputs))   # 这句话不会报错
print(net(a[0]))     # 在不修改数据类型时，此处报错

后来又看到一篇类似的文章：
https://blog.csdn.net/weixin_42419002/article/details/89200285