expected scalar type Long but found Float

Pytorch tensor と numpy ndarray の変換 - Pythonいぬ (hatenablog.com)

 

主要还是 array 转 tensor的时候要确定类型

x = torch.from_numpy(x.astype(np.float32))