Pytorch Shape 1