【数据结构与算法】PyTorch中permute与contiguous对tensor结构进行变换

本文介绍PyTorch中contiguous()函数的作用及其使用场景,通过一个实例展示了如何将张量调整为内存中连续分布的形式,并进一步进行view操作。文章详细解释了在使用如transpose或permute等函数后为何需要调用contiguous()。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

contiguous()——把tensor变成在内存中连续分布的形式

需要变成连续分布的情况:

contiguous:view只能用在contiguous的variable上。如果在view之前用了transpose, permute等,需要用contiguous()来返回一个contiguous copy。

import torch
import numpy as np

def main():
    
    myArr = np.array([[[1,2,3],
                      [4,5,6]]])
    print(myArr.shape)
    print(myArr.size)
    myTsr = torch.tensor(myArr)
    print(myTsr)
    print(myTsr.shape)
    myTsr2 = myTsr.permute(2,0,1)
    print(myTsr2)
    print(myTsr2.shape)
    myTsr3 = myTsr2.contiguous().view(-1,2)
    print(myTsr3)
    print(myTsr3.shape)

if __name__ == '__main__':
    main()

运行结果:

 

(1, 2, 3)
6
tensor([[[1, 2, 3],
         [4, 5, 6]]], dtype=torch.int32)
torch.Size([1, 2, 3])
tensor([[[1, 4]],

        [[2, 5]],

        [[3, 6]]], dtype=torch.int32)
torch.Size([3, 1, 2])
tensor([[1, 4],
        [2, 5],
        [3, 6]], dtype=torch.int32)
torch.Size([3, 2])

 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值