这是别人总结好的点击打开链接
下面是我自己做的一些测试
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)