torch.stack(list)

最新推荐文章于 2025-07-11 16:22:28 发布

摇摆的果冻

最新推荐文章于 2025-07-11 16:22:28 发布

阅读量9.5k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： Pytorch

本文链接：https://blog.youkuaiyun.com/Z_lbj/article/details/85012764

Pytorch 专栏收录该内容

12 篇文章

订阅专栏

本文详细介绍了PyTorch中使用stack函数堆叠张量的方法，包括如何通过指定不同维度来组合张量，提供了具体示例并展示了输出结果。

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

torch.stack(list,0)

list 中的每个元素为tensor 中第0维度的每个元素

import torch

a = torch.Tensor([[1, 3, 2], [1, 3, 2]])
b = torch.Tensor([[2, 1, 1], [2, 1, 1]])
c = torch.Tensor([[3, 2, 3], [2, 1, 1]])

my_list = [a, b, c]
print(torch.stack(my_list, 0))


》》》
tensor([[[ 1.,  3.,  2.],
         [ 1.,  3.,  2.]],

        [[ 2.,  1.,  1.],
         [ 2.,  1.,  1.]],

        [[ 3.,  2.,  3.],
         [ 2.,  1.,  1.]]])

troch.stack(list,1) list中每个元素的第0维元素-第n维元素各成一组

print(torch.stack(my_list, 1))

》》》
tensor([[[ 1.,  3.,  2.],
         [ 2.,  1.,  1.],
         [ 3.,  2.,  3.]],

        [[ 1.,  3.,  2.],
         [ 2.,  1.,  1.],
         [ 2.,  1.,  1.]]])