In some cases, it can be crucial to return the output as is within your console, such as when you want to repeat-paste the output. My conclusion is the fact even if you want to have the additional dimension of torch.stack, utilizing torch.cat after which reshape is best. @baptiste It has, but if you need to write numerous strains passing the