not sure if you're aware of this one, but its pretty sweet. Probably identical on performance regardless:

```python
yield np.split(sampled, array([num_x]))
```