Attention-Based Mechanisms for Cognitive Reinforcement Learning