分类：Numpy扩充矩阵维度

2021
03-23

详解Numpy扩充矩阵维度(np.expand_dims, np.newaxis)和删除维度(np.squeeze)的方法

在操作矩阵的时候，不同的接口对于矩阵的输入维度要求不同，输入可能为1-D，2-D，3-D等等。下面介绍一下使用Numpy进行矩阵维度变更的相关方法。主要包括以下几种：1、np.newaxis扩充矩阵维度2、np.expand_dims扩充矩阵维度3、np.squeeze删除矩阵中维度大小为1的维度np.newaxis，np.expand_dims扩充矩阵维度：importnumpyasnpx=np.arange(8).reshape(2,4)print(x.shape)#添加第0维,输出shape->(1,2,4)x1=x[np.newaxi... 继续阅读 >