torch.cat()
多个张量连结(concatenate)在一起,把它们端对端地叠起来形成一个更大的张量。我们只需要提供张量列表,并给出沿哪个轴连结。
下面的例子分别演示了当我们沿行(轴-0,形状的第一个元素)和按列(轴-1,形状的第二个元素)连结两个矩阵时,会发生什么情况。
我们可以看到:
第一个输出张量的轴-0长度(6)是两个输入张量轴-0长度的总和(3+3);
第二个输出张量的轴-1长度(8)是两个输入张量轴-1长度的总和(4+4)。
torch.cat(张量,dim = 0/1)