src/transformers/models/rembert/modeling_tf_rembert.py/TFRemBertForTokenClassification/__init__
class TFRemBertForTokenClassification:    def __init__(self, config: RemBertConfig, *inputs, **kwargs):
        super().__init__(config, *inputs, **kwargs)

        self.num_labels = config.num_labels

        self.rembert = TFRemBertMainLayer(config, name="rembert", add_pooling_layer=False)
        self.dropout = tf.keras.layers.Dropout(rate=config.hidden_dropout_prob)
        self.classifier = tf.keras.layers.Dense(
            units=config.num_labels, kernel_initializer=get_initializer(config.initializer_range), name="classifier"
        )