src/transformers/models/esm/modeling_tf_esm.py/TFEsmContactPredictionHead/build
class TFEsmContactPredictionHead:    def build(self, input_shape):
        super().build(input_shape)
        with tf.name_scope("regression"):
            self.regression.build((None, self.in_features))