def METHOD_NAME(
    self,
    x: torch.Tensor,
    label: torch.Tensor,
    label_mask: torch.BoolTensor = None,
    position_encoding: torch.Tensor = None,
    attention_mask: torch.LongTensor = None,
    unique_name: List[str] = None,
    **kwargs,
):
    return self._general_forward(
        x, label, label_mask, position_encoding, attention_mask, unique_name
    )