Added Visualizer, fixed bug

Added the visualizer class, which lets you see all of the attention heads.

Also fixed a bug where calculated the E and F matrices. They were calculated to be (n,d), but instead, they should have been (n,k). This has since been fixed.