pytorch每日一学48(torch.vstack())垂直(行方向)按顺序堆叠张量

第48个方法

torch.vstack(tensors, *, out=None) → Tensor

此方法就是按垂直方向堆叠，可与
torch.hstack()进行对比查看，torch.hstack就是按水平方向堆叠张量。

直接看例子，很好理解：

>>> a = torch.tensor([1, 2, 3])
>>> b = torch.tensor([4, 5, 6])
>>> torch.vstack((a,b))
tensor([[1, 2, 3],
        [4, 5, 6]])
>>> a = torch.tensor([[1],[2],[3]])
>>> b = torch.tensor([[4],[5],[6]])
>>> torch.vstack((a,b))
tensor([[1],
        [2],
        [3],
        [4],
        [5],
        [6]])

其实就是将所有的行拼起来就可以了