https://researchdatapod.com/pytorch-cat-vs-stack-explained/
PyTorch Cat Vs Stack Explained