src/transformers/models/xlnet/modeling_tf_xlnet.py/TFXLNetLMHead/build
class TFXLNetLMHead:    def build(self, input_shape):
        self.bias = self.add_weight(shape=(self.vocab_size,), initializer="zeros", trainable=True, name="bias")
        super().build(input_shape)