src/transformers/models/groupvit/modeling_tf_groupvit.py/TFGroupViTStage/concat_x
class TFGroupViTStage:    def concat_x(self, x: tf.Tensor, group_token: Optional[tf.Tensor] = None) -> tf.Tensor:
        if group_token is None:
            return x
        return tf.concat([x, group_token], axis=1)