code: def _intensity_aug(key, x, scale=0.05):
      """Follows the code in Schwarzer et al. (2020) for intensity augmentation."""
      r = jax.random.normal(key, shape=(x.shape[0], 1, 1, 1))
      noise = 1.0 + (scale * jnp.clip(r, -2.0, 2.0))
      return x * noise
  
imports: import copy
  import functools
  from absl import logging
  from dopamine.jax import networks
  from dopamine.jax.agents.full_rainbow import full_rainbow_agent
  import gin
  import jax
  import jax.numpy
  import numpy
  import tensorflow
label_desc: Use image node_id 4 for calling a global function with example usage: _intensity_aug(key, x, scale) and returns: unknown