class_docstr: Interface for all policy networks in EvoJAX.
class_docstr_tok: ["Interface","for","all","policy","networks","in","EvoJAX","."]
docstr: Get vectorized actions.
  
  Args:
      t_states - Task states.
      params - A batch of parameters, shape is (num_envs, param_size).
      p_states - Policy internal states.
  Returns:
      jnp.ndarray. Vectorized actions.
      PolicyState. Internal states.
docstr_tok: ["Get","vectorized","actions",".","Args",":","t_states","-","Task","states",".","params","-","A","batch","of","parameters",",","shape","is","(","num_envs",",","param_size",")",".","p_states","-","Policy","internal","states",".","Returns",":","jnp.ndarray",".","Vectorized","actions",".","PolicyState",".","Internal","states","."]
code: def get_actions(
      self,
      t_states: TaskState,
      params: jnp.ndarray,
      p_states: PolicyState,
  ) -> Tuple[jnp.ndarray, PolicyState]:
      """Get vectorized actions.
  
      Args:
          t_states - Task states.
          params - A batch of parameters, shape is (num_envs, param_size).
          p_states - Policy internal states.
      Returns:
          jnp.ndarray. Vectorized actions.
          PolicyState. Internal states.
      """
      raise NotImplementedError()
  
file_path: evojax/evojax/policy/base.py
imports: from abc import ABC
  from abc import abstractmethod
  from typing import Tuple
  import jax.numpy
  import jax.random
  from flax.struct import dataclass
  from evojax.task.base import TaskState
label_desc: Use image node_id 2 for calling the PolicyNetwork obj's underlying member method code with example usage: obj.get_actions(t_states, params, p_states) without return types