ENH: `create_diagonal`: support broadcasting

Thus, the docstring would be something like:
```
    """A vectorized version of diagonal.

    Args:
        m: Has shape (*k, n)
    Returns: Array with shape (*k, n, n) and the elements of m on the diagonals.
    """
```
If this is okay, I can submit a PR