[学习笔记]torch x = x.view(x.size(0),-1)的理解

1.先定义一个Tensor，输出该内容和大小

import torch
 
a = torch.arange(1,17)
 
print(a)
print(a.size(0))

 2.Tensor转为一维

import torch
 
a = torch.arange(1,17)
 
print(a)
print(a.size(0))
 
b = a.view(a.size(0),-1)
print(b)

 其中这个-1指的是不知道多少列的情况下，根据原来Tensor内容和Tensor的大小自动分配列数

效果等同于：

 
c = a.view(a.size(0),1)
print(c)

4.我们还可以将行数不定（为-1），指定列数为Tensor的大小：

import torch
 
a = torch.arange(1,17)
 
print(a)
print(a.size(0))
 
b = a.view(a.size(0),-1)
print(b)
c = a.view(a.size(0),1)
print(c)
 
d = a.view(-1, a.size(0))
print(d)