Splitting image into small batches

[ptrblck]

I think unfold should work. This post gives you an example.