在将数组放入网络之前,有时需要增加维度,此时只需要加[None]即可实现
>>> a=np.array([[1,1,1],[1,1,1]])
>>> a
#array([[[1, 1, 1],
[1, 1, 1]]])
>>> a.shape #(2, 3)
增加维度
>>> a[None].shape #[1,2,3] 加在第一维度
>>> a[None]
#array([[[1, 1, 1],
[1, 1, 1]]])
在将数组放入网络之前,有时需要增加维度,此时只需要加[None]即可实现
>>> a=np.array([[1,1,1],[1,1,1]])
>>> a
#array([[[1, 1, 1],
[1, 1, 1]]])
>>> a.shape #(2, 3)
增加维度
>>> a[None].shape #[1,2,3] 加在第一维度
>>> a[None]
#array([[[1, 1, 1],
[1, 1, 1]]])