Topics:
Layer construction function for an identity layer.
Based on jax.example_libraries.stax.Identity.
Tuple[InitFn, ApplyFn, LayerKernelFn]
Tuple
InitFn
ApplyFn
LayerKernelFn
(init_fn, apply_fn, kernel_fn).