pytorch張量拼接怎么實(shí)現(xiàn)

小億
102
2024-01-09 12:46:22

在PyTorch中，可以使用torch.cat()函數(shù)來(lái)實(shí)現(xiàn)張量的拼接。

torch.cat()函數(shù)的語(yǔ)法如下：

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

其中，參數(shù)tensors是一個(gè)張量的序列，表示要拼接的張量；dim是指定拼接的維度，默認(rèn)為0（沿著行的方向拼接）；out是一個(gè)可選的輸出張量，表示拼接的結(jié)果。

下面是一個(gè)使用torch.cat()函數(shù)進(jìn)行張量拼接的示例代碼：

import torch

# 創(chuàng)建兩個(gè)張量
tensor1 = torch.tensor([[1, 2, 3], [4, 5, 6]])
tensor2 = torch.tensor([[7, 8, 9], [10, 11, 12]])

# 沿著行的方向拼接張量
result = torch.cat((tensor1, tensor2), dim=0)

print(result)

運(yùn)行結(jié)果為：

tensor([[ 1,  2,  3],
        [ 4,  5,  6],
        [ 7,  8,  9],
        [10, 11, 12]])

在上述示例中，我們首先創(chuàng)建了兩個(gè)張量tensor1和tensor2。然后，通過(guò)torch.cat()函數(shù)將這兩個(gè)張量沿著行的方向進(jìn)行拼接，得到了一個(gè)新的張量result。最后，我們打印出了拼接結(jié)果。

0 踩