Localized Sliding Attention Mechanism - Coding problem | OneNoughtOne