张量堆叠函数torch.stack()

📚博客主页：knighthood2001
✨公众号：认知up吧 （目前正在带领大家一起提升认知，感兴趣可以来围观一下）
🎃知识星球：【认知up吧|成长|副业】介绍
❤️感谢大家点赞👍🏻收藏⭐评论✍🏻，您的三连就是我持续更新的动力❤️
🙏笔者水平有限，欢迎各位大佬指点，相互学习进步！

torch.stack 是 PyTorch 中用于在一个新的维度上堆叠张量序列的函数。当你有多个张量，想要将它们按照某个维度进行堆叠时，就可以使用 torch.stack。

具体来说，torch.stack 接受一个张量序列作为输入，并在指定的维度上堆叠这些张量，生成一个新的张量。在这个过程中，要求所有的输入张量的形状必须是一致的，除了沿着堆叠维度的尺寸之外。新生成的张量的形状将会在指定的维度上增加一个维度。

下面是 torch.stack 的基本语法：

torch.stack(tensors, dim=0)

其中：

• tensors 是一个张量序列，即一个张量列表或元组。
• dim 是指定的维度，表示在哪个维度上进行堆叠。默认值为 0。

举个例子，假设有两个形状相同的张量 tensor1 和 tensor2，形状为 (3, 2)，如果我们想在新的维度上将它们堆叠起来，可以这样做：

stacked_tensor = torch.stack([tensor1, tensor2], dim=0)

这样，stacked_tensor 的形状将会是 (2, 3, 2)，其中第一个维度是堆叠维度，表示堆叠后的张量序列的数量。