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