Add more details about the output distogram format.
Add more details about the output distogram format.

PiperOrigin-RevId: 310134498

deepmind/deepmind-research,deepmind/deepmind-research,deepmind/deepmind-research,deepmind/deepmind-research