Pytorch实现List Tensor转Tensor,reshape拼接等操作

放养的密瓜

持续更新一些常用的Tensor操作，比如List，Numpy，Tensor之间的转换，Tensor的拼接，维度的变换等操作。
其它Tensor操作如 einsum等见：待更新。
用到两个函数：

• 1. torch.cat

  复制代码
• 1. torch.stack

  复制代码

一、List Tensor转Tensor (torch.cat)

1. // An highlighted block
   
2. >>> t1 = torch.FloatTensor([[1,2],[5,6]])
   
3. >>> t2 = torch.FloatTensor([[3,4],[7,8]])
   
4. >>> l = []
   
5. >>> l.append(t1)
   
6. >>> l.append(t2)
   
7. >>> ta = torch.cat(l,dim=0)
   
8. >>> ta = torch.cat(l,dim=0).reshape(2,2,2)
   
9. >>> tb = torch.cat(l,dim=1).reshape(2,2,2)
   
10. >>> ta
    
11. tensor([[[1., 2.],
    
12.          [5., 6.]],
    
13. 
    
14.         [[3., 4.],
    
15.          [7., 8.]]])
    
16. >>> tb
    
17. tensor([[[1., 2.],
    
18.          [3., 4.]],
    
19. 
    
20.         [[5., 6.],
    
21.          [7., 8.]]])

复制代码

高维tensor

** 如果理解了2D to 3DTensor,以此类推，不难理解3D to 4D，看下面代码即可明白：**

1. >>> t1 = torch.range(1,8).reshape(2,2,2)
   
2. >>> t2 = torch.range(11,18).reshape(2,2,2)
   
3. >>> l = []
   
4. >>> l.append(t1)
   
5. >>> l.append(t2)
   
6. >>> torch.cat(l,dim=2).reshape(2,2,2,2)
   
7. tensor([[[[ 1.,  2.],
   
8.           [11., 12.]],
   
9. 
   
10.          [[ 3.,  4.],
    
11.           [13., 14.]]],
    
12. 
    
13. 
    
14.         [[[ 5.,  6.],
    
15.           [15., 16.]],
    
16. 
    
17.          [[ 7.,  8.],
    
18.           [17., 18.]]]])
    
19. >>> torch.cat(l,dim=1).reshape(2,2,2,2)
    
20. tensor([[[[ 1.,  2.],
    
21.           [ 3.,  4.]],
    
22. 
    
23.          [[11., 12.],
    
24.           [13., 14.]]],
    
25. 
    
26. 
    
27.         [[[ 5.,  6.],
    
28.           [ 7.,  8.]],
    
29. 
    
30.          [[15., 16.],
    
31.           [17., 18.]]]])
    
32. >>> torch.cat(l,dim=0).reshape(2,2,2,2)
    
33. tensor([[[[ 1.,  2.],
    
34.           [ 3.,  4.]],
    
35. 
    
36.          [[ 5.,  6.],
    
37.           [ 7.,  8.]]],
    
38. 
    
39. 
    
40.         [[[11., 12.],
    
41.           [13., 14.]],
    
42. 
    
43.          [[15., 16.],
    
44.           [17., 18.]]]])

复制代码

二、List Tensor转Tensor (torch.stack)

代码：

1. import torch
   
2. 
   
3. t1 = torch.FloatTensor([[1,2],[5,6]])
   
4. t2 = torch.FloatTensor([[3,4],[7,8]])
   
5. l = [t1, t2]
   
6. 
   
7. t3 = torch.stack(l, dim=2)
   
8. print(t3.shape)
   
9. print(t3)
   
10. 
    
11. ## output:
    
12. ## torch.Size([2, 2, 2])
    
13. ## tensor([[[1., 3.],
    
14. ##          [2., 4.]],
    
15. ##        [[5., 7.],
    
16. ##         [6., 8.]]])

复制代码

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。

来源:https://www.jb51.net/article/266622.htm
免责声明：由于采集信息均来自互联网，如果侵犯了您的权益，请联系我们【E-Mail:cb@itdo.tech】 我们会及时删除侵权内容，谢谢合作！