一、拼接張量
拼接（Concatenation）張量是將兩個(gè)張量沿著某個(gè)維度進(jìn)行拼接，得到一個(gè)更大的張量。在PyTorch中，可以使用torch.cat來完成拼接張量的操作。
import torch
# 創(chuàng)建3 x 2的張量
x = torch.randn(3, 2)
# 創(chuàng)建3 x 3的張量
y = torch.randn(3, 3)
# 沿著第二個(gè)維度對(duì)兩個(gè)張量進(jìn)行拼接
z = torch.cat([x, y], dim=1)
print(z)
在上面的例子中，我們先使用torch.randn創(chuàng)建了兩個(gè)不同的張量x和y，張量x的維度是3 x 2，張量y的維度是3 x 3。使用torch.cat將張量x和y沿著第二個(gè)維度（即列）拼接，得到了一個(gè)維度為3 x 5的新張量z。
二、注意事項(xiàng)
在使用torch.cat進(jìn)行張量拼接時(shí)，需要注意以下幾點(diǎn)。
拼接的維度的大小必須相同，除拼接維度外，其他維度大小也必須相同。 拼接的維度的編號(hào)必須在0到張量維度數(shù)減1的范圍內(nèi)。 拼接的維度大小可以根據(jù)需要設(shè)置為-1，此時(shí)大小將自動(dòng)推斷。 如果兩個(gè)張量是CPU張量，則拼接后的張量也是CPU張量。如果兩個(gè)張量是CUDA張量，則拼接后的張量也是CUDA張量。三、拼接多個(gè)張量
我們也可以使用torch.cat來拼接多個(gè)張量。下面的例子將展示如何同時(shí)拼接三個(gè)張量。
import torch
# 創(chuàng)建3 x 2的張量
x = torch.randn(3, 2)
# 創(chuàng)建3 x 3的張量
y = torch.randn(3, 3)
# 創(chuàng)建3 x 4的張量
z = torch.randn(3, 4)
# 沿著第二個(gè)維度對(duì)三個(gè)張量進(jìn)行拼接
w = torch.cat([x, y, z], dim=1)
print(w)
在上面的例子中，我們分別創(chuàng)建了3個(gè)不同大小的張量，使用torch.cat將它們沿著第二個(gè)維度（即列）拼接成一個(gè)維度為3 x 9的張量w。
四、使用stack拼接張量
如果需要在新創(chuàng)建的維度上拼接張量，可以使用torch.stack。棧（Stack）張量是一個(gè)新的張量，它將輸入張量沿著新創(chuàng)建的維度進(jìn)行堆疊。
import torch
# 創(chuàng)建3 x 2的張量
x = torch.randn(3, 2)
# 創(chuàng)建3 x 2的張量
y = torch.randn(3, 2)
# 沿著新維度將兩個(gè)張量進(jìn)行堆疊
z = torch.stack([x, y], dim=0)
print(z)
在上面的例子中，我們先使用torch.randn創(chuàng)建了兩個(gè)不同的張量x和y，張量x和張量y的維度都是3 x 2。使用torch.stack將張量x和張量y沿著新維度（即第0個(gè)維度）堆疊，得到了一個(gè)維度為2 x 3 x 2的新張量z。
五、結(jié)論
在PyTorch中，torch.cat和torch.stack是非常有用的函數(shù)，它們可以方便地對(duì)多個(gè)張量進(jìn)行拼接操作。在使用這兩個(gè)函數(shù)時(shí)需要注意維度的大小和編號(hào)，以及張量的類型。
京公網(wǎng)安備 11010802030320號(hào)