The way to indicate shape is
```a : array_like, shape (..., N)
```
I used array_like here instead of array as I assume nested lists and such will be converted. Is that so?
Needs fixing in the rest of the documentation.