torch.stack
- torch.stack(tensors, dim=0, *, out=None) Tensor
Concatenates a sequence of tensors along a new dimension.
All tensors need to be of the same size.
Concatenates a sequence of tensors along a new dimension.
All tensors need to be of the same size.