code: def embedding(
      scope: Scope,
      num_embeddings: int,
      features: int,
      init_fn=default_embed_init,
  ) -> Embedding:
      """Creates embedding dataclass.
  
      Args:
        num_embeddings: number of embeddings.
        features: Number of feature dimensions for each embedding.
        embedding_init: embedding initializer.
  
      Returns:
        Embedding dataclass with lookup and attend methods.
      """
      table = scope.param("table", init_fn, (num_embeddings, features))
      return Embedding(table)
  
file_path: flax/flax/core/nn/linear.py
imports: from collections.abc import Iterable
  import jax.numpy
  import numpy
  from jax import lax
  from flax import struct
  from flax.core import Scope
  from flax.linen import initializers
label_desc: Use image node_id 7 for calling a global function with example usage: embedding(scope, num_embeddings, features, init_fn) and returns: Embedding