1、说明:
reshape()和numpy.reshape()函数的作用是,重塑的数组的shape。
2、注意:(参考链接1:Python中reshape函数参数-1的意思?)
python默认是按行取元素。
参数-1,表示模糊reshape的意思。
比如:reshape(-1,3),固定3列 多少行不知道。
3、实验代码:
1 import numpy as np 2 import torch 3 a = [[1,2,3],[4,5,6]] #定义一个数组 4 b = torch.tensor(a) #初始化一个tensor 5 print("原始数组为:\n{0}\n\n".format(b)) 6 print("reshape(-1),(就是把任何shape的tensor拉平):\n{0}\n\n".format(b.reshape(-1)))#string.format(var)是python里的格式化输出函数 7 print("reshape(-1,2),(reshape成三行两列):\n{0}\n\n".format(b.reshape(-1,2)))
4、实验结果(jupyter notebook)