fix `torch.jit.trace_module` documentation (#40248)
Summary:
This should fix https://github.com/pytorch/pytorch/issues/39328
Before:
After:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/40248
Reviewed By: ezyang
Differential Revision: D22195038
Pulled By: zou3519
fbshipit-source-id: c4bff6579a422a56ed28b644f5558b20d901c94e
