def METHOD_NAME(self, inputs: torch.Tensor) -> EncoderOutputDict:
    """
    Params:
        inputs: The inputs fed into the encoder.
                Shape: [batch x vocab_size], type tf.int32.
    Returns:
        Embeddings of shape [batch x vocab_size x embed size], type float32.
    """
    hidden = self.embed(inputs)
    hidden = self.fc_stack(hidden)
    return {ENCODER_OUTPUT: hidden}