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)