深度学习

Pytorch中cat和stack的区别

Pytorch中cat和stack的区别

Pytorch中cat和stack的区别

cat和stack都是PyTorch中的函数,用于将多个张量(tensor)沿着某个维度(dim)连接起来。它们的区别在于,cat不会改变张量的维数,而stack会增加一个新的维度。例如,如果有两个形状为(2,3)的张量a和b,那么:

  • cat(a,b,dim=0)会得到一个形状为(4,3)的张量c,相当于将a和b在第0维上拼接起来。

  • stack(a,b,dim=0)会得到一个形状为(2,2,3)的张量d,相当于在a和b之间增加了一个新的维度,然后将它们在这个维度上拼接起来。

评论

10:24:00
8月27日, 2026, 周三
system_stats.sh
处理器 CPU82%
运行内存 RAM12.4G / 16G
网络吞吐 Network5.2 Mbps · 4g

精选图库

JacyLunatic

JacyLunatic

只争朝夕

91 篇创作

实时天气

24°
本地
28° / 24°
97% 12 km/h 体感 29°
逐小时
多日预报

创作活跃度

萌ICP备20241011号
本站已加入BLOGS·CN