print(torch.arange(start=1,end=9))
print(torch.reshape(torch.arange(start=1,end=10),shape=[3,1,3])) print(torch.reshape(torch.arange(start=1,end=10),shape=[1,3,3])) print(torch.reshape(torch.arange(start=1,end=10),shape=[3,3,1]))
在reshape中,第一位默认是需要修改类型的数据,shape输入的是修改后数据的维度,第一个最高维,依次类推,
以第一个reshape代码举例,简单的来说,shape中第一个数字是3,那么最大的方括号中就会有三个同级的方括号,第二个数字是1,所以第二级的方括号中只有一个最低级的方括号,最后一个数字是3,所以最低级的方括号里会有三个数字,同理剩下的两也是一样的道理。