torch.stack()

最新推荐文章于 2025-05-23 19:45:13 发布

原创最新推荐文章于 2025-05-23 19:45:13 发布 · 659 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#pytorch #深度学习 #机器学习

收录于

本文详细介绍了PyTorch中的stack函数用法，通过示例展示了如何沿新维度拼接两个张量，并解释了拼接前后张量形状的变化规律。

'''
#torch.stack((a,b),dim)沿着新的dim拼接a,b,大前提就是：a,b的形状要完全一样,拼接前a,b是三维的那么
#拼接后a,b就是四维的
from this import d
import torch
ysh=torch.rand((3,4,5))
mes=torch.zeros((3,4,5))
kaka=torch.stack((ysh,mes),dim=1)
print("kaka_shape",kaka.shape)
#kaka_shape  dim
#[2, 3, 4, 5] 0
#[3, 2, 4, 5] 1
'''