pytorch合并拆分运算

最新推荐文章于 2025-01-02 19:00:00 发布
原创 最新推荐文章于 2025-01-02 19:00:00 发布 · 361 阅读 · 0 ·
CC 4.0 BY-SA版权
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
文章标签：

#pytorch #python #深度学习

文章介绍了PyTorch中处理张量的几种关键运算，包括Cat用于沿指定维度拼接张量，要求非拼接维度一致；stack则是在张量的现有维度间插入新维度，所有维度需相同；而split函数用于拆分张量，指定尺寸和维度即可进行分割。这些操作在处理多维数据如图像通道时尤为重要。

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

pytorch合并拆分运算

简述

  • 如果维度少了直接添加一个维度
  • 如果每一个维度的偏少，直接扩展每一个维度的元素数量

在这里插入图片描述

Cat合并操作

在这里插入图片描述

在这里插入图片描述

cat操作的维度必须是一致的，只能有一个维度不一样（拼接的维度），但是其他的维度都必须是一样的

案例：为一张RGB照片再添加一个通道变成四通道

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-HVFxDedf-1686399780061)(../images/1291c2fa7790c1978bedac952b7042bba6b9c24b3b41164cbe00416f8156bdbb.png)]

stack-插入新的维度操作

stack操作两个张量的维度必须全部保持一致

在这里插入图片描述

stack操作是创建一个新的维度

splict拆分操作

第一个参数是拆分的尺寸，第二个参数是拆分的维度
在这里插入图片描述

pytorch中图像的分块（patch）操作（使用了.permute()维度转换）
sinat_42239797的博客
03-05 1万+
参考：1.https://www.zhihu.com/question/60321866 2. https://blog.csdn.net/appleml/article/details/80143212?depth_1-utm_source=distribute.pc_relevant.none-task&utm_source=distribute.pc_relevant.none-ta...
Pytorch入门教程（四）：Tensor拆分、合并及基本运算
小明同学
07-01 1万+
1. cat 进行维度拼接 a = torch.rand(4, 32, 8) b = torch.rand(5, 32, 8) c = torch.cat([a, b], dim=0) # 按第0维度进行拼接，除拼接之外的维度必须相同 print(c.shape) 结果：torch.Size([9, 32, 8]) 2. stack 产生一个新的维度 a = torch.rand...
pytorch合并与拆分
qq_42178122的博客
08-09 242
第一： broading cast a = torch.randn(4,3) b = torch.rand(4,3) c = a + b c = torch.randn(1,3) d = a+ c 结果： a: tensor([[-0.9123, -0.3065, -0.5646], [ 1.4327, 2.1845, -1.6093], [ 0.9051, -1.2144, -1.6093], [-0.6509, 0.7420, 1.0669
图像切割-pytorch
weixin_53226226的博客
04-12 867