python pytorch view()函數

最新推荐文章于 2025-04-29 08:30:00 发布

xidaoliang123

最新推荐文章于 2025-04-29 08:30:00 发布

阅读量1.1k

点赞数 3

分类专栏：学习笔记

本文链接：https://blog.youkuaiyun.com/xidaoliang/article/details/106078275

版权

学习笔记专栏收录该内容

28 篇文章

订阅专栏

view是改变tensor的形状，并不只是简单的只是多行tensor拼接成一行，view中的-1是自适应的调整

A = torch.arange(0, 16)
print(A)
#tensor([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14, 15])
B = A.view(-1,2)
print(B)
#tensor([[ 0,  1],
        [ 2,  3],
        [ 4,  5],
        [ 6,  7],
        [ 8,  9],
        [10, 11],
        [12, 13],
        [14, 15]])
B = A.view(2,-1)
print(B)
#tensor([[ 0,  1,  2,  3,  4,  5,  6,  7],
        [ 8,  9, 10, 11, 12, 13, 14, 15]])
C = A.view(-1,2,2)
print(C)
#tensor([[[ 0,  1],
         [ 2,  3]],

        [[ 4,  5],
         [ 6,  7]],

        [[ 8,  9],
         [10, 11]],

        [[12, 13],
         [14, 15]]])
C = A.view(2,2,-1)
print(C)
#tensor([[[ 0,  1,  2,  3],
         [ 4,  5,  6,  7]],

        [[ 8,  9, 10, 11],
         [12, 13, 14, 15]]])