class_docstr: Base class of solver.
class_docstr_tok: ["Base","class","of","solver","."]
docstr: Get the input function.
  return a Task class
docstr_tok: ["Get","the","input","function",".","return","a","Task","class"]
code: def input_fn(self, mode):
      """Get the input function.
      return a Task class
      """
      task_name = self.config["data"]["task"]["name"]
      self._task = registers.task[task_name](self.config, mode)
      return self._task
  
imports: import abc
  import math
  import delta.compat
  import tensorflow_addons
  from absl import logging
  from delta import utils
  from delta.utils import optimizer
  from delta.utils.register import registers
label_desc: Use image node_id 3 for calling the Solver obj's underlying member method code with example usage: obj.input_fn(mode) and returns: self