np.newaxis 可以帮助我们为数组增加一个新维度,比如说,将一个向量转换成列矩阵和行矩阵:
首先导入numpy
import numpy as np
后续代码如下:
高维数组增加维度:
增加维度也可以用np.expand_dims