Improve docs for init_rpc (#40298)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/40298
Test Plan: Imported from OSS
Differential Revision: D22143155
Pulled By: mrshenli
fbshipit-source-id: deadcc29eda157b401ca6a091c3ba17455acb6b5