Task-Context-Aware Imitation Learning