torch.vstack — PyTorch 1.10.1 documentation
pytorch.org › docs › stabletorch.vstack. torch.vstack(tensors, *, out=None) → Tensor. Stack tensors in sequence vertically (row wise). This is equivalent to concatenation along the first axis after all 1-D tensors have been reshaped by torch.atleast_2d (). Parameters. tensors ( sequence of Tensors) – sequence of tensors to concatenate.