[numpy]增加或者减少一维
增加维度使用np.newaxis
,减少维度使用np.squeeze
np.newaxis
>>> a = np.arange(3)
>>> a.shape
(3,)
# 增加第一维
>>> b = a[np.newaxis, :]
>>> b.shape
(1, 3)
np.squeeze
# 减少第一维
>>> c = b.squeeze(0)
>>> c.shape
(3,)
增加维度使用np.newaxis
,减少维度使用np.squeeze
>>> a = np.arange(3)
>>> a.shape
(3,)
# 增加第一维
>>> b = a[np.newaxis, :]
>>> b.shape
(1, 3)
# 减少第一维
>>> c = b.squeeze(0)
>>> c.shape
(3,)