Convert tokenizer outputs for Keras in doc example
The TF examples in `training.mdx` don't turn the tokenizer outputs into a `dict`, so Keras gets confused.

Fixes #20709 
The docs for this PR live [here](https://moon-ci-docs.huggingface.co/docs/transformers/pr_20732). All of your documentation changes will be reflected on that endpoint.