code: def hybrid_forward(self, F, words, weight):
          
          #pylint: disable=arguments-differ
          embeddings = F.sparse.dot(words, weight)
          return embeddings
docs: Compute embedding of words in batch.
  
          Parameters
          ----------
          words : mx.nd.NDArray
              Array of token indices.