train_x = reshape(train_x, [size(train_x,1) 1 28 28]);
时间: 2023-10-26 13:05:22 浏览: 86
队列实现火车厢重排的算法及代码(个人创作)
5星 · 资源好评率100%
这行代码的作用是将训练数据的形状从原本的 [n_samples, 28, 28] 转换成 [n_samples, 1, 28, 28],其中 n_samples 是训练数据的数量。这是因为在神经网络中,输入数据的形状通常是四维的,包括样本数量、通道数、高度和宽度。因此,将训练数据的通道数从 1 开始是为了适应卷积神经网络等模型的输入要求。具体来说,这里将原本的二维图像数据扩展成了一个四维数组,第一维是样本数量,第二维是通道数(在这里是 1),第三维是图像的高度,第四维是图像的宽度。
阅读全文