`mask_non_zeros` seems better to me.