代码中遇到的numpy函数

np.expand_dims(a,axis=0)

该函数是对数组a进行维度扩充
其中a是一个多维数组，axis=0，1，2表示分别在第1,2，3个位置扩充, 即插入一维

a = [[1,2,3],[4,5,6]]
c1 = np.expand_dims(a,axis=0)
c2 = np.expand_dims(a,axis=1)
c3 = np.expand_dims(a,axis=2)
print('*********c1**********')
print(c1,c1.shape)
print('*********c2**********')
print(c2,c2.shape)
print('**********c3*********')
print(c3,c3.shape)

得到结果

*********c1**********
[[[1 2 3]
  [4 5 6]]] (1, 2, 3)
*********c2**********
[[[1 2 3]]

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

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

np.concatenate((a,b),axis = *)

将数组或者列表a,b在axis的方向上拼接起来

a = [[1,2,3],[4,5,6]]
c = np.concatenate([a]*2,1)
print(c)

得到结果

axis = 0