美文网首页
PyTorch view()

PyTorch view()

作者: 坐看云起时zym | 来源:发表于2020-07-08 11:25 被阅读0次

    PyTorch中的view函数可以改变tensor的形状

    eg1

    import torch
    test = torch.zeros(2,3)
    print(test)
    print(test.view(3,-1)) #-1表示程序会根据上一个维度自动计算-1所在位置的维度
    
    #tensor([[0., 0., 0.],
    #        [0., 0., 0.]])
    
    #tensor([[0., 0.],
    #        [0., 0.],
    #        [0., 0.]])
    

    eg2

    test = torch.Tensor([1,2,3,4,5,6,7,8])
    print(test)
    print(test.view(4,2))
    
    # tensor([1., 2., 3., 4., 5., 6., 7., 8.])
    
    #tensor([[1., 2.],
    #        [3., 4.],
    #        [5., 6.],
    #        [7., 8.]])
    

    相关文章

      网友评论

          本文标题:PyTorch view()

          本文链接:https://www.haomeiwen.com/subject/vfiqcktx.html