def METHOD_NAME(input, mask):
    """masked_select
    Slice the value from given Mask
    Args:
        input: Input tensor to be selected
        mask: A bool tensor for sliced.
    Return:
        Part of inputs where mask is True. 
    """
    index = L.where(mask)
    return L.gather(input, index, overwrite=False)