view()
参数及返回值
view(*shape) → Tensor
功能
返回一个与原Tensor内容,元素数量也相同但是形状不同的Tensor。
举例
# 将(1, 6)尺寸的Tensor转换为尺寸为(3, 2)
>>> import torch
>>> tt1=torch.tensor([-0.3623, -0.6115, 0.7283, 0.4699, 2.3261, 0.1599])
>>> tt2=tt1.view(3,2)
>>> tt2
tensor([[-0.3623, -0.6115],
[ 0.7283, 0.4699],
[ 2.3261, 0.1599]])
# 将(3, 2)尺寸的Tensor转换为尺寸为(2, x), -1 表示做自适应性调整,
# 如果参数只有一个-1,则表示Tensor的第一维的维度默认为1
>>> tt3=tt2.view(2,-1)
>>> tt3
tensor([[-0.3623, -0.6115, 0.7283],
[ 0.4699, 2.3261, 0.1599]])
viewas(ExampleTensor)
功能
将原Tensor的Siz转换成ExampleTensor的Size。
举例
# 接上面
>>> tt4=tt2.viewas(tt1)
>>> tt4
tensor([-0.3623, -0.6115, 0.7283, 0.4699, 2.3261, 0.1599])
参考资料
官网解读
Pytorch-view的用法