I suggest we'd rename it into `batch_mask` or something along those lines. @JohanMabille , what do you think?