numpy库函数numpy.newaxis的使用
numpy 以下简称np, np.newaxis的作用是增加原来数组的维度,比如原来是一个二维数组,通过使用np.newaxis可以变成一个三维数组。同理一维数组可以变成二维数组,三维数组可以变成四维。
形状为(5,)的一维数组通过增加坐标轴变成二维数组
# 示例
# 一维数组变成二维,且增加的坐标轴位置不同
import numpy as np
import torch as t
pt ...
原创
2020-03-12 23:02:26 ·
364 阅读 ·
0 评论