Python PyTorch column_stack用法及代碼示例

本文簡要介紹python語言中 torch.column_stack 的用法。

用法:

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

參數：

tensors(張量序列) -要連接的張量序列

關鍵字參數：

out(Tensor,可選的) -輸出張量。

通過在 tensors 中水平堆疊張量來創建一個新張量。

等效於 torch.hstack(tensors) ，除了 tensors 中的每個零維或一維張量 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]])