Graph-based Neural Modules to Inspect Attention-based Architectures: A Position Paper