快捷方式

torch.column_stack

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

通过水平堆叠 tensors 中的张量来创建一个新张量。

等效于 torch.hstack(tensors),除了 tensors 中的每个零维或一维张量 t 首先被重塑为 (t.numel(), 1) 列,然后水平堆叠。

参数

tensors (序列 of 张量) – 要连接的张量序列

关键字参数

out (张量, 可选) – 输出张量。

示例

>>> 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 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源