[Docs] Improve PyTorch, Flax generate API (#15988)
* Move generate docs
* up
* Update docs/source/_toctree.yml
* correct
* correct some stuff
* correct tests
* more fixes
* finish generate
* add to doc stest
* finish
* finalize
* add warning to generate method