Occasionally, it is important to return the output as is inside your console, as an example when you want to copy-paste the output. My conclusion is even if you'd like to have the additional dimension of torch.stack, employing torch.cat then reshape is healthier. @baptiste It's got, but if you want https://www.petskyonline.com/