Pytorch Cnn Code Example