一、函数简介
Pytorch中的view函数主要用于Tensor维度的重构,即返回一个有相同数据但不同维度的Tensor。
根据上面的描述可知,view函数的操作对象应该是Tensor类型。如果不是Tensor类型,可以通过tensor = torch.tensor(data)来转换。
二、实例讲解
▶view(参数a,参数b,…),其中,总的参数个数表示将张量重构后的维度。
import torch
temp = [1,2,3,4,5,6] # temp的类型为list,非Tensor
temp = torch.tensor(temp) # 将temp由list类型转为Tensor类型
print(temp) # torch.Size([6])
print(temp.view(2,3)) # 将temp的维度改为2*3
print(temp.view(2,3,1)) # 将temp的维度改为2*3*1
print(temp.view(2,3,1,1)) # 更多的维度也没有问题,只要保证维度改变前后的元素个数相同就行,即2*3*1*1=6
▶view(参数a,参数b,…),其中,如果某个参数为-1,则表示该维度取决于其它维度,由Pytorch自己补充。
import torch
temp = [[11,12,13,14,15,16],
[21,22,23,24,25,26]]
temp = torch.tensor(temp)
print(temp)
# torch.Size([2, 6])
print(temp.view(3,-1,2))
# 这里的-1表示该维度取决于其它维度,即等于(2*6)÷3÷2=2
# torch.Size([3, 2, 2])
▶view(-1)表示将Tensor转为一维Tensor。
import torch
temp = [1,2,3,4,5,6] # temp的类型为list,非Tensor
temp = torch.tensor(temp) # 将temp由list类型转为Tensor类型
print(temp) # 本身就是一维张量
print(temp.view(-1)) # 因此,转变后还是一维,没什么变换
temp1 = torch.tensor([[1,2,3],[4,5,6]])
print(temp1) # torch.Size([2, 3])
print(temp1.view(-1)) # 多维张量转为一维张量
文章来源:https://www.toymoban.com/news/detail-596417.html
欢迎各位伙伴们在评论区交流!文章来源地址https://www.toymoban.com/news/detail-596417.html
到了这里,关于详解Pytorch中的view函数的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!