Passing constants which depend on input to custom RNN cell in TF2.x?

Question

TF1.x earlier had legacy support for Recurrent class which had the get_constants function which was a callback which could be used to write functio