Multi-Task Reinforcement Learning with Context-based Representations