How to convert LSTMCell to tensorflow or keras

nlp
1

I wan to convert pytorch LSTMCell to keras or tensorflow. Anyone can help?
For example, self.att_lstm = nn.LSTMCell(1536, 512)
h_att, c_att = self.att_lstm(att_lstm_input, ((2,512), 2,512))