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.
© 2019 Torch Contributors
Licensed under the 3-clause BSD License.
https://pytorch.org/docs/1.8.0/generated/torch.stack.html