Update loss module doc

[ssnl]

Fixes #{issue number}

[dr-ci]

💊 CI failures summary and remediations

As of commit 14d66fb (more details on the Dr. CI page):

---

• 1/2 failures possibly* introduced in this PR
  • 1/1 non-CircleCI failure(s)
• 1/2 broken upstream at merge base 0b66cda on Nov 30 from 6:48am to 3:14pm

---

🚧 1 fixed upstream failure:

These were probably caused by upstream breakages that were already fixed.

Please rebase on the viable/strict branch (expand for instructions)

If your commit is older than viable/strict, run these commands:

git fetch https://github.com/pytorch/pytorch viable/strict
git rebase FETCH_HEAD

Check out the recency history of this "viable master" tracking branch.

• pytorch_bazel_build on Nov 30 from 6:48am to 3:14pm (5bb2a87 - d74f2d2)
  • 🔁 rerun

---

This comment was automatically generated by Dr. CI (expand for details).

Follow this link to opt-out of these comments for your Pull Requests.

Please report bugs/suggestions to the (internal) Dr. CI Users group.

[zou3519]

Can we get a before and after screenshot to make sure the reduce didn't disappear?

[facebook-github-bot]

@zou3519 has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.

[zou3519]

Built locally, seems fine to me:

[facebook-github-bot]

@zou3519 merged this pull request in d2e96fc.