多语言展示
当前在线:447今日阅读:126今日分享:42

python中np.rollaxis的使用

在深度学习中,当把tensor数据转换成numpy数据,然后将numpy数据的维度进行调整位置,经常会用到该命令。
工具/原料
1

python

2

numpy

方法/步骤
1

numpy中的rollaxis的函数的用法如下如所示

2

记住这个用法,一个简单的诀窍就是将start轴的位置放置在axis的后面。

3

如下图的程序,就是将轴0放置在轴的后面

4

具体的分析如下图所示,数组的下标是从0,1,2转变成1,2,0

推荐信息