Pytorch Shape 1900