Reshape是一种数据有效处理方式,可以使数据变得更加紧凑且具有意义。PyTorch中torch.reshape函数可以改变张量的形状,相当于是对张量中的元素重新排列。这个函数的参数有两个: 其中input是需要转换形状的张量,shape是要变换的具体形状,可以是tuple或list类型。函数返回一个张量,它的形状将会与shape参数中指定的形状相同。
torch.reshape(),如下图所示: 2. torch.reshape()函数解析 torch.reshape(input,shape)→ Tensor 返回将输入的形状转变为shape指定的形状大小,元素总数不变。 3.代码举例 a = torch.zeros(size=(5,4)) b = a.reshape(-1)#输出张量b的size为torch.Size([20]) c = a.reshape(2,-1) #输出张量c的si...