def hybrid_forward(self, F, words, weight):
        """Compute embedding of words in batch.

        Parameters
        ----------
        words : mx.nd.NDArray
            Array of token indices.

        """
        #pylint: disable=arguments-differ
        embeddings = F.sparse.dot(words, weight)
        return embeddings