Fix visualization of CNN.
Merge pull request #114 from AlexanderFabisch/fix-mnist-visualization

Fix visualization of CNN.