import torch
a = torch.tensor([[1,2,3],[4,5,6]])
print(a.size())
b = torch.tensor([[7,8,9]])
print(b.size())
c = torch.cat((a,b),dim=0)
print(c.size())
print(c)
torch.Size([2, 3])
torch.Size([1, 3])
torch.Size([3, 3])
tensor([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
import torch
a = torch.tensor([[1,2],[5,6]])
print(a.size())
b = torch.tensor([[7,8,9],[10,11,12]])
print(b.size())
c = torch.cat((a,b),dim=1)
print(c.size())
print(c)
torch.Size([2, 2])
torch.Size([2, 3])
torch.Size([2, 5])
tensor([[ 1, 2, 7, 8, 9],
[ 5, 6, 10, 11, 12]])
torch.cat(inputs,dim)
imputs:要合并的tensor。要求剔除dim维度之后,所有tensor的形状要相同。
dim:所有tensor要在dim维度上合并,tensor 在dim上的数不一定要一致。