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

目录

一、List Tensor转Tensor (torch.cat)

高维tensor

二、List Tensor转Tensor (torch.stack)

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

torch.cattorch.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.]],
14.         [[3., 4.],
15.          [7., 8.]]])
16. >>> tb
17. tensor([[[1., 2.],
18.          [3., 4.]],
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.]],
10.          [[ 3.,  4.],
11.           [13., 14.]]],
13.         [[[ 5.,  6.],
14.           [15., 16.]],
16.          [[ 7.,  8.],
17.           [17., 18.]]]])
18. >>> torch.cat(l,dim=1).reshape(2,2,2,2)
19. tensor([[[[ 1.,  2.],
20.           [ 3.,  4.]],
22.          [[11., 12.],
23.           [13., 14.]]],
25.         [[[ 5.,  6.],
26.           [ 7.,  8.]],
28.          [[15., 16.],
29.           [17., 18.]]]])
30. >>> torch.cat(l,dim=0).reshape(2,2,2,2)
31. tensor([[[[ 1.,  2.],
32.           [ 3.,  4.]],
34.          [[ 5.,  6.],
35.           [ 7.,  8.]]],
37.         [[[11., 12.],
38.           [13., 14.]],
40.          [[15., 16.],
41.           [17., 18.]]]])

复制代码

二、List Tensor转Tensor (torch.stack)

代码：

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

复制代码

以上为个人经验，希望能给大家一个参考，也希望大家多多支持中国红客联盟。