首先创建一个数组,维度是[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数组的排列方式其实没有变，变得只是括号，也就是[]的位置
数组改变维度
猜你喜欢
转载自blog.csdn.net/Gussss/article/details/90180968
今日推荐
周排行