ptional*):
                Used if `summary_type == "cls_index"` and takes the last token of the sequence as classification token.

        Returns:
            `jnp.array`: The summary of the sequence hidden states.
        Nr