Should `self.initializer.check_mask(shape)` be called?