class_docstr: Identity layer, convenient for giving a name to an array.
class_docstr_tok: ["Identity","layer",",","convenient","for","giving","a","name","to","an","array","."]
code: def __call__(self, x: jnp.ndarray) -> jnp.ndarray:
      return x
  
imports: import math
  from dopamine.discrete_domains import atari_lib
  from flax import linen
  import gin
  import jax.numpy
label_desc: Use image node_id 1 for calling the IdentityLayer obj's underlying member method code with example usage: obj.__call__(x) and returns: x