I confess I don't really understand what is going on here, particularly this bit ` _mask.shape = result.shape`. Is this some sort of broadcasting? What is the relation of the two iterators? I could study the code, but the easier route is to ask you ;) Maybe a comment would be helpful.