from . import functions


def psa_mask(input, psa_type=0, mask_H_=None, mask_W_=None):
    return functions.psa_mask(input, psa_type, mask_H_, mask_W_)