code: def add_gaussian_noise(
      key: Array, action: Array, stddev: float
  ) -> Array:
      """Returns continuous action with noise drawn from a Gaussian distribution.
  
      Args:
        key: a key from `jax.random`.
        action: continuous action scalar or vector.
        stddev: standard deviation of noise distribution.
  
      Returns:
        noisy action, of the same shape as input action.
      """
      chex.assert_type(action, float)
  
      noise = jax.random.normal(key, shape=action.shape) * stddev
      return action + noise
  
file_path: rlax/rlax/_src/exploration.py
imports: from typing import Optional, Union
  import chex
  import jax
  import jax.numpy
  from rlax._src import episodic_memory
label_desc: Use image node_id 1 for calling a global function with example usage: add_gaussian_noise(key, action, stddev) and returns: unknown