Add docstring.

```
'''
Create a diagonal array section for this extent.

If the extent does not lie on the diagonal, a zero array is returned.
Args:
  array (DistArray):
  ex (Extent): Region being processed.
'''