Pytroch中的view()&viewas()函数

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的用法

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章