numpy 改变ndarray维数的一些操作
np.expand_dims(a, axis)
--扩展array的形状,在指定的axis维度插入新的一维,输入a是array
例子:
>>>x = np.array([1,2])
>>>x.shape
(2,)
>>>y = np.expand_dims(x, axis=0)
>>>y
array([[1,2]])
>>>y.shape
(1,2)
>>>y = np.expand_dims(x, axis=1)
>>>y
array([[1],
[2]])
>>>y.shape
(2,1)
numpy.squeeze(a, axis=None)
--移除array中维数为1的维度,axis可以指定移除维数为1的某个维度,返回移除这些维度之后的array
实例:
>>>x = np.array([[[0], [1], [2]]]
>>>x.shape
(1, 3, 1)
>>>np.squeeze(x).shape
(3,)
>>>np.squeeze(x, axis=0).shape
(3,1)
所以squeeze和expand_dims一对互逆的操作
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步