[DOCS] Fix the docs for the inputs arg of trace_module func (#41586)
Summary:
Fix the docs for the `inputs` arg of `trace_module` func.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/41586
Reviewed By: ezyang
Differential Revision: D22598453
Pulled By: zou3519
fbshipit-source-id: c2d182238b5a51f6d0a7d0683372d72a239146c5