Visualization tool for deep learning