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

时间：2020-07-26 01:24:35 收藏：0 阅读：114

矩阵增加一个维度：

将矩阵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)

A: (1,3,1)