python 矩阵增加/减少一个维度

矩阵增加一个维度：

将矩阵A： m×n×p 转化为  m×n×p×1 以实现多维矩阵叠加的效果:

import numpy as np
A = np.array([[1, 2, 3]])
B =A [:, : ,np.newaxis]

结果：

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,)