class_docstr: MPO loss for a categorical policy (Abdolmaleki et al., 2018).
  
  (Abdolmaleki et al., 2018): https://arxiv.org/pdf/1812.02256.pdf
class_docstr_tok: ["MPO","loss","for","a","categorical","policy","(","Abdolmaleki","et","al.",",","2018",")",".","(","Abdolmaleki","et","al.",",","2018",")",":","https",":","\/\/arxiv.org\/pdf\/1812.02256.pdf"]
docstr: Creates an initial set of parameters.
docstr_tok: ["Creates","an","initial","set","of","parameters","."]
code: def init_params(self, action_dim: int, dtype: DType = jnp.float32):
      """Creates an initial set of parameters."""
      del action_dim  # Unused.
      return CategoricalMPOParams(
          log_temperature=jnp.full(
              [1], self._init_log_temperature, dtype=dtype
          ),
          log_alpha=jnp.full([1], self._init_log_alpha, dtype=dtype),
      )
  
file_path: acme/acme/agents/jax/mpo/categorical_mpo.py
imports: from typing import NamedTuple, Tuple
  import distrax
  import jax
  import jax.numpy
label_desc: Use image node_id 2 for calling the CategoricalMPO obj's underlying member method code with example usage: obj.init_params(action_dim, dtype) and returns: CategoricalMPOParams