版权声明:转载请注明出处及原文地址。 https://blog.csdn.net/zl1085372438/article/details/86635323
torch.stack(sequence, dim=0, out=None),做tensor的拼接。sequence表示Tensor列表,dim表示拼接的维度,注意这个函数和concatenate是不同的,torch的concatenate函数是torch.cat,是在已有的维度上拼接,而stack是建立一个新的维度,然后再在该纬度上进行拼接。
torch.stack
a = torch.rand(3, 4)
b = torch.rand(3, 4)
d = torch.stack((a, b), dim=0)
print(a.shape)
print(b.shape)
print(d.shape)
---------------------------------------------
torch.Size([3, 4])
torch.Size([3, 4])
torch.Size([2, 3, 4])
torch.cat
a=torch.Tensor([[1,2,3]])
b=torch.Tensor([[7,8,9]])
d=torch.cat( (a,b) ,dim = 0)
print(a.shape)
print(b.shape)
print(d.shape)
---------------------
torch.Size([1, 3])
torch.Size([1, 3])
torch.Size([2, 3])
原文:https://blog.csdn.net/th_num/article/details/81979346