同样都要指定维度
cat指定维度是原维度相加,在原维度上相加,不增加维度,指定的dim为增加维度的位置
stack指定维度是根据数量得到结果,增加一个维度,值为叠的元素数量
eg.
a = torch.rand((4, 3))
b = torch.rand((4, 3))
c = torch.stack((a, b), dim=-1)
>>>c.shape = (4, 3, 2) #这个2是因为是2个元素
d = torch.cat((a,b), dim=0)
>>>d.shape = (8, 3)# 8=4+4