pytorch view 用法

view函数主要用于Tensor维度的重构，即返回一个有相同数据但不同维度的Tensor。

import torch

data = [[1,2,3], [4,5,6], [7,8,9]] # data的类型为list，非tensor

data = torch.tensor(data) # 将data由list类型转为tensor类型

print(data) # torch.Size([6])

#print(data.view(data.size(0), -1))

print(data.size())
print(data.size(0))

print(data.view(3, 3, -1))

# 确定一个维度，-1 的维度会被自动计算