Fix trace docs (#24191)
Summary:
These were incorrect and didn't run before
Pull Request resolved: https://github.com/pytorch/pytorch/pull/24191
Pulled By: driazati
Differential Revision: D16770604
fbshipit-source-id: 0d8547185871f7f4b1e44c660e45699ed8240900
