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: Get action sampled from the policy.
  
  Args:
      observation (np.ndarray): Augmented observation from the
          environment, with shape :math:`(O+N, )`. O is the dimension of
          observation, N is the number of tasks.
  
  Returns:
      np.ndarray: Action sampled from the policy,
          with shape :math:`(A, )`. A is the dimension of action.
      dict: Action distribution information.
docstr_tok: ["Get","action","sampled","from","the","policy",".","Args",":","observation","(","np.ndarray",")",":","Augmented","observation","from","the","environment",",","with","shape",":","math",":","`","(","O+N",",",")","`",".","O","is","the","dimension","of","observation",",","N","is","the","number","of","tasks",".","Returns",":","np.ndarray",":","Action","sampled","from","the","policy",",","with","shape",":","math",":","`","(","A",",",")","`",".","A","is","the","dimension","of","action",".","dict",":","Action","distribution","information","."]
code: def get_action(self, observation):
      """Get action sampled from the policy.
  
      Args:
          observation (np.ndarray): Augmented observation from the
              environment, with shape :math:`(O+N, )`. O is the dimension of
              observation, N is the number of tasks.
  
      Returns:
          np.ndarray: Action sampled from the policy,
              with shape :math:`(A, )`. A is the dimension of action.
          dict: Action distribution information.
  
      """
  
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 7 for calling the TaskEmbeddingPolicy obj's underlying member method code with example usage: obj.get_action(observation) without return types