rse_coo` argument.
        If `sparse_coo=True`, then the mask is stored as sparse coo else dense tensor
        r!