(Python)Numpy矩阵增加/减少一个维度

Section 1:矩阵增加一个维度

方法一:

使用np.newaxis将m*n维矩阵转化为m*n*1维矩阵。

(Python)Numpy矩阵增加/减少一个维度_第1张图片

 

方法二:

使用np.expand_dims来增加一个新的维度。

(Python)Numpy矩阵增加/减少一个维度_第2张图片

Link:

https://numpy.org/doc/stable/reference/generated/numpy.expand_dims.html?highlight=expand%20dim#numpy.expand_dims 

Section 2:矩阵减小一个维度

使用np.squeeze来减少矩阵中多余的维度。

(Python)Numpy矩阵增加/减少一个维度_第3张图片

Link:

 https://numpy.org/doc/stable/reference/generated/numpy.squeeze.html?highlight=squeeze#numpy.squeeze

你可能感兴趣的:(笔记,矩阵,python,线性代数)