Numpy --expand_dims

这是别人总结好的点击打开链接

下面是我自己做的一些测试

a = np.array([[1, 2, 3], [4, 5, 6]])
print(a.shape)
print(a)
print('\n')

b = np.expand_dims(a, 0)
print(b)
print(b.shape)
print('\n')
c = np.expand_dims(a, 1)
print(c)
print(c.shape)
print('\n')
d = np.expand_dims(a, 2)
print(d)
print(d.shape)
print('\n')
e = np.expand_dims(a, 3)
print(e)
print(e.shape)

结果：

(2, 3)
[[1 2 3]
 [4 5 6]]




[[[1 2 3]
  [4 5 6]]]
(1, 2, 3)




[[[1 2 3]]


 [[4 5 6]]]
(2, 1, 3)




[[[1]
  [2]
  [3]]


 [[4]
  [5]
  [6]]]
(2, 3, 1)




[[[1]
  [2]
  [3]]


 [[4]
  [5]
  [6]]]
(2, 3, 1)