[numpy]增加或者减少一维

参考:numpy.ndarray 增加一维

增加维度使用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,)