numpy newaxis与expand_dims

numpy.newaxis 为数组增加一个新维度,机器学习的常用框架中处理的数据大多是矩阵,经常需要将一个向量转换成列矩阵或者行矩阵;

同样numpy.expand_dims也可以实现对应的功能,效果如下:

猜你喜欢

转载自blog.csdn.net/szfhy/article/details/82147586