align minibatches

[ferrine]

sometimes rngs are not aligned, so this function can help

[twiecki]

LGTM.

[gBokiau]

Hi guys,

I'm presuming that in the presence of multiple minibatches in a model, this would allow the same rows to be sampled at each iteration.

Any chance for a quick write up on how this might work?

In particular, is this supposed to be called during model definition, or as a callback of sorts?

[gBokiau]

From what I'm gathering this should best be called right before fitting, as model construction (again, purely guessing what I'm observing) might break alignments if some minibatches are used more than once and/or altered, dimshuffled or otherwise transformed in the model declaration.

[ferrine]

Yes, alignment should be done right before any action that involves them. No need to rerun notebook after occasional minibatch. eval()