数组改变维度

首先创建一个数组,维度是[2,3,4,5] import torch import numpy as np a=np.array(range(120)).reshape(2,3,4,5)  可以看到数字是依次排列的,从最后一个维度,也就是站从[5]这个维度慢慢开始填充,填充了5个元素之后,开始扩散到[4]这个维度。 b=a.reshape(5,4,3,2) 可以看到b和a数组的排列方式其实没有变,
相关文章
相关标签/搜索