Pytorch Shape 172nd