我使用python / Numpy将小图像存储到ndarray中。 当我试图将ndarray从32,32,1形状转换为1,32,32,1时，我就陷入了困境。有帮助吗？感谢
答案 0 :(得分：0)
您需要扩展numpy数组的尺寸。使用np.expand_dims。
arr = np.expand_dims(arr, axis=0)
答案 1 :(得分：0)
arr[np.newaxis, :, :, :]将有效
答案 2 :(得分：0)
除了显式添加轴，您还可以显式地重新整形它以添加轴：
>>> import numpy as np
>>> arr = np.ones((32, 32, 1)) # just ones for demonstration purposes
>>> reshaped = arr.reshape(1, *arr.shape)
>>> reshaped.shape
(1, 32, 32, 1)