Hi,
torch.cat 1.5k concatenates along an existing dimension. and so the number of dimensions of the output is the same as the inputs.
torch.stack 765 stacks a list of tensors along an new dimension. In that case the number of dimensions of the output is one more than the inputs.
I think you’re looking for the second in your case.