for _ in range(num_layers):
      self.layers.append(
          TransformerEncoder(