Python PyTorch cat用法及代碼示例

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

用法:

torch.cat(tensors, dim=0, *, out=None) → Tensor

參數：

• tensors(張量序列) -任何相同類型的張量的python序列。提供的非空張量必須具有相同的形狀，cat 維度除外。

• dim(int,可選的) -張量連接的維度

關鍵字參數：

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

連接給定維度中的給定 seq 張量序列。所有張量必須具有相同的形狀(連接維度除外)或為空。

torch.cat() 可以看作是 torch.split() 和 torch.chunk() 的逆運算。

torch.cat() 可以通過示例得到最好的理解。

例子：

>>> x = torch.randn(2, 3)
>>> x
tensor([[ 0.6580, -1.0969, -0.4614],
        [-0.1034, -0.5790,  0.1497]])
>>> torch.cat((x, x, x), 0)
tensor([[ 0.6580, -1.0969, -0.4614],
        [-0.1034, -0.5790,  0.1497],
        [ 0.6580, -1.0969, -0.4614],
        [-0.1034, -0.5790,  0.1497],
        [ 0.6580, -1.0969, -0.4614],
        [-0.1034, -0.5790,  0.1497]])
>>> torch.cat((x, x, x), 1)
tensor([[ 0.6580, -1.0969, -0.4614,  0.6580, -1.0969, -0.4614,  0.6580,
         -1.0969, -0.4614],
        [-0.1034, -0.5790,  0.1497, -0.1034, -0.5790,  0.1497, -0.1034,
         -0.5790,  0.1497]])