Implement `transformer_internals_module_1_problem_4` in Python. Test cases verify edge cases and functional requirements.
transformer_internals_module_1_problem_4(*args, **kwargs)