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]])