由torch.cat()引发的维度问题

                                         *

由torch.cat()引发的维度问题

	说实话,我看了CSDN的好几篇这个问题的博客只说到二维的,但是我们搞图像的。处理数据起步都要三维了(RGB),另外网上一种言论的关于什么0代表行拼接,1代表列拼接,我就不做过多评价了,我先声明我是小菜鸡,但是总结的东西基本得到我师兄的认可的。你也可以再琢磨琢磨。
	进入正题:

1:torch.cat((x0,x1),dim)
语义解释一下,torch.cat是拼接数据的一个模块(不知道这么说对不对,反正是实现这个功能)。第一个部分就是(,)输入要拼接的数据,第二个部分就是输入要拼接的维度。
import torch
a0 = torch.Tensor([[[[1,1,1],[2,2,2]]]])
a1 = torch.Tensor([[[[3,3,3],[4,4,4]]]])
x = torch.cat((a0,a1),3).type(torch.FloatTensor)
#拼接对象是数据a0和数据a1,维度是三维的。
print(x)
2.数据维度是什么?
a0 = torch.Tensor([[[[1,1,1],[2,2,2]]]])
a1 = torch.Tensor([[[[3,3,3],[4,4,4]]]])
判断一个张量是几维的,看看它前面有几个方括号就可以了,这是最简便的方法了。数一数。本例子中a0就是一个4维的张量。
3.拼接维度是什么?
import torch
a0 = torch.Tensor([[[[1,1,1],[2,2,2]]]])
a1 = torch.Tensor([[[[3,3,3],[4,4,4]]]])
x = torch.cat((a0,a1),3).type(torch.FloatTensor)
我们再用cat实现拼接时就要填写后面的dim数值,这里的维度是怎么看的呢?
[[[1,1,1],[2,2,2]]]——————这个就是第0维(3个方括号,原数据是4维张量)
[[1,1,1],[2,2,2]]——————这个就是第1维(2个方括号,原数据是4维张量)
[1,1,1],[2,2,2]——————这个就是第2维(1个方括号,原数据是4维张量)
1,1,1,2,2,2———————这个就是第3维(0个方括号,原数据是4维张量)
他们分别的拼接结果如下:
0维:
tensor([[[[1., 1., 1.],
[2., 2., 2.]]],
[[[3., 3., 3.],
[4., 4., 4.]]]
])
1维:
tensor([[[[1., 1., 1.],
[2., 2., 2.]],
[[3., 3., 3.],
[4., 4., 4.]]
]])
2.维:
tensor([[[[1., 1., 1.],
[2., 2., 2.],
[3., 3., 3.],
[4., 4., 4.]
]]])
3维:
tensor([[[[1., 1., 1., 3., 3., 3.],
[2., 2., 2., 4., 4., 4.]
]]]
如果你还有不理解的可以给我留言。

  • 13
    点赞
  • 30
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 15
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 15
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

少卿不在大理寺

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值