Pytorch Example Cnn