class_docstr: Module for embedding tokens in a low-dimensional space.
code: def __call__(self, inputs):
      if self.densify_gradients:
          embeddings = dense_gradient(self.embeddings)
      else:
          embeddings = self.embeddings
      return tf.nn.embedding_lookup(embeddings, inputs)
  
file_path: sonnet/sonnet/src/embed.py
imports: import math
  from typing import Optional
  from sonnet.src import base
  from sonnet.src import initializers
  from sonnet.src import types
  import tensorflow
label_desc: Use image node_id 2 for calling the Embed obj's underlying member method code with example usage: obj.__call__(inputs) and returns: tf