In some instances, it is crucial to return the output as is in the console, for example when you need to repeat-paste the output. My conclusion is that even if you need to have the extra dimension of torch.stack, employing torch.cat and after that reshape is healthier. We shall see. https://nielsont752pyg1.activosblog.com/profile