Pytorch Shape 1 Side