向mnist数据集中添加元素时报错

Question

打开图像implus, 转为灰度图，大小28*28
添加出错，报错如下

Answer 1

你的train_images维度是(60000, 28, 28)，不可以直接+(28,28)啊，你可以下面这样操作

# 使用numpy.expand_dims将样本添加到数组的最后一维
expanded_array = np.expand_dims(implus, axis=0)

# 将扩展后的数组拼接到原数组的末尾
train_images = np.concatenate((train_images, expanded_array), axis=0)

# 现在train_images的形状是(60001, 28, 28)