Pytorch Read Image In Python