class_docstr: Module for embedding tokens in a low-dimensional space.
class_docstr_tok: ["Module","for","embedding","tokens","in","a","low-dimensional","space","."]
code: def embeddings(self):
      return hk.get_parameter(
          "embeddings",
          [self.vocab_size, self.embed_dim],
          init=self.w_init,
      )
  
file_path: dm-haiku/haiku/_src/embed.py
imports: from collections.abc import Sequence
  import enum
  from typing import Optional, Union
  from haiku._src import base
  from haiku._src import initializers
  from haiku._src import module
  import jax
  import jax.numpy
  import numpy
label_desc: Use image node_id 2 for calling the Embed obj's underlying member method code with example usage: obj.embeddings() and returns: hk