python 矩陣增加/減少一個維度

矩陣增加一個維度：

將矩陣A： m×n×p 轉化為  m×n×p×1 以實現多維矩陣疊加的效果:

import numpy as np
A = np.array([[1, 2, 3]])
B =A [:, : ,np.newaxis]
或者：
B = np.expand_dims(A, axis=2) # 增加一個維度

結果：

A: (1,3)

B: (1,3,1)

矩陣減少冗余維度

import numpy as np
a = [[[1, 2, 3]]]
a = np.array(a)
B = np.squeeze(a)

結果：

A: (1,1,3)

B: (3,)