pytorch中的view函数

2024年5月13日 281次阅读来源: 王二的石锅拌饭

import torch
 
a = torch.Tensor(2,3)
print(a)
# tensor([[0.0000, 0.0000, 0.0000],
#        [0.0000, 0.0000, 0.0000]])
 
print(a.view(1,-1))
# tensor([[0.0000, 0.0000, 0.0000, 0.0000, 0.0000, 0.0000]])

view函数相当于resize的功能，将原来的tensor变换成指的维度

    原文作者：王二的石锅拌饭
    原文地址: https://zhuanlan.zhihu.com/p/70140517
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。