class_docstr: Base class for Task Embedding policies in TensorFlow.
  
  This policy needs a task id in addition to observation to sample an action.
class_docstr_tok: ["Base","class","for","Task","Embedding","policies","in","TensorFlow",".","This","policy","needs","a","task","id","in","addition","to","observation","to","sample","an","action","."]
docstr: akro.Box: One-hot space of task id.
docstr_tok: ["akro.Box",":","One-hot","space","of","task","id","."]
code: def task_space(self):
      """akro.Box: One-hot space of task id."""
      return self.encoder.spec.input_space
  
file_path: garage/src/garage/tf/policies/task_embedding_policy.py
imports: import abc
  from garage.tf.policies.policy import Policy
label_desc: Use image node_id 4 for calling the TaskEmbeddingPolicy obj's underlying member method code with example usage: obj.task_space() and returns: self