Fix example block format in Distributed Optimizer API doc (#34919)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/34919
Test Plan: Imported from OSS
Differential Revision: D20500013
Pulled By: mrshenli
fbshipit-source-id: d28cbdd1ec207e1e8501ce389b7040fb764f12ca