def forward(self, input_ids):
    (embedding_output, features) = self.embeddings(input_ids)
    (encoded, attn_weights) = self.encoder(embedding_output)
    return (encoded, attn_weights, features)
def forward(self, input_ids):
    (embedding_output, features) = self.embeddings(input_ids)
    (encoded, attn_weights) = self.encoder(embedding_output)
    return (encoded, attn_weights, features)