pytorch view 用法

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))

发表回复

您的电子邮箱地址不会被公开。