增加数据维度 new_data = np.expand_dims(data, axis=int)
参数axis表示:在哪里加一个维度,其对应的是data.shape中的位置索引(axis<=len(data.shape)),
0表示最前,-1表示最末。
假设原始数据 data.shape = (28,28,1),则:
axis = 0时:(28,28,1)🔜 (1,28,28,1)
axis = 1时:(28,28,1)🔜 (28,1,28,1)
axis = -1时:(28,28,1)🔜 (28,28,1,1)
axis = 4时:会报错,超出了原来的最大维度个数
也可以使用reshape方法达到相同的目的,例如:
data.reshape((data.shape[0], data.shape[1], data.shape[2], 1))
# 就等价与
np.expand_dims(data, axis=-1)