import torch


def score(size:int, mask:torch.Tensor) -> torch.Tensor:
    """
    Count number of makred (with False) elements.

    Note, 1D mask is reshaped into (size, -1)

    Parameters
    ----------
    size: int
        reshape size
    mask: torch.Tensor
        mask

    Returns
    -------
    score (torch.Tensor)

    """
    return mask.logical_not().reshape(size, -1).sum(-1)