mlr_context_torch function

Context for Torch Learner