src/transformers/models/layoutlm/modeling_tf_layoutlm.py/TFLayoutLMPooler/call
class TFLayoutLMPooler:    def call(self, hidden_states: tf.Tensor) -> tf.Tensor:
        # We "pool" the model by simply taking the hidden state corresponding
        # to the first token.
        first_token_tensor = hidden_states[:, 0]
        pooled_output = self.dense(inputs=first_token_tensor)

        return pooled_output