Pytorch View 1 1