神经网络 pytorch 分类二维矩阵图像和一维向量的方法