Inserting custom layer after embeddings layer in BERT


I want to add a fully connected dense layer of dimension 256, right after the initial BERT embedding layer, i.e. between the embedding layer and the first encoder layer. Is there an easier way to go about this, other than making changes in the code of the original BertModel class?