Pytorch文档学习 TORCH.COLUMN_STACK

最新推荐文章于 2026-07-06 11:29:05 发布

原创最新推荐文章于 2026-07-06 11:29:05 发布 · 308 阅读

0 ·

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

收录于

pytorch文档自学习

当前文章被收录于：

pytorch文档自学习

9 篇文章 0 人学习

订阅专栏查看详情

当前文章被以下社区和专栏收录：

torch.column_stack函数用于将多个张量沿列方向堆叠。它接受一组张量作为输入，将每个零维或一维张量转换为列向量后进行堆叠。例如，给定两个一维张量a和b，它们会被堆叠成一个二维张量，形成一个新的列。对于多维张量，如b，它们的形状会被调整以匹配列堆叠的要求。

torch.column_stack

torch.column_stack(tensors, *, out=None) → Tensor

把两个tensor纵向堆叠在一起。相当于 torch.hstack (张量) ，但是张量中的每个零维或一维张量 t 在水平叠加之前首先被重新形成一个(t.numel () ，1)列。

a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5, 6])
torch.column_stack((a, b))

tensor([[1, 4],
    [2, 5],
    [3, 6]])

a = torch.arange(5)
b = torch.arange(10).reshape(5, 2)
torch.column_stack((a, b, b))

tensor([[0, 0, 1, 0, 1],
        [1, 2, 3, 2, 3],
        [2, 4, 5, 4, 5],
        [3, 6, 7, 6, 7],
        [4, 8, 9, 8, 9]])

标签

#pytorch #学习 #深度学习