def METHOD_NAME(self, mask):
    mask_3d_btd = mask[:, :, None]
    if len(mask.shape) == 2:
        mask_4d_bhlt = 1 - mask[:, None, None, :]
    elif len(mask.shape) == 3:
        mask_4d_bhlt = 1 - mask[:, None, :]
    mask_4d_bhlt = mask_4d_bhlt * -10000.0
    return mask_3d_btd, mask_4d_bhlt