Update the docstrings of the newly refactored TransformerEncoder
Update the docstrings of the newly refactored TransformerEncoder