A very good point - that should be trivial to implement with `np.broadcast_arrays`.  Will add a note about this in the version I'll push up in a bit