code: def unnormalize(
      state: PopArtState, normalized: Array, indices: Array
  ) -> Array:
      """Returns unnormalized values.
  
      Args:
        state: The PopArt summary stats.
        normalized: normalized values that we apply PopArt to.
        indices: Which scale and shifts to use
  
      Returns:
        Unnormalized PopArt values.
      """
      scale = state.scale[indices]
      shift = state.shift[indices]
      unnormalized = scale * normalized + shift
      return unnormalized
  
file_path: rlax/rlax/_src/pop_art.py
imports: import collections
  from typing import Mapping, Tuple
  import chex
  import jax
  import jax.numpy
  from rlax._src import base
label_desc: Use image node_id 3 for calling a global function with example usage: unnormalize(state, normalized, indices) and returns: unnormalized