преобразование ячейки LSTM из модели pytroch в keras

Question

Я написал код для ячейки LSTM в pytorch и той же модели, которую хочу построить в keras, через s подкласс (с помощью метода вызова __ init__, def). Может ли кто-нибудь помочь мне решить эту проблему?

Код Pytorch:

класс MyModel (torch.nn.Module):

def __init__(self,hidden_size):
    super(MyModel,self).__init__()
    self.lstm = torch.nn.LSTMCell(10,hidden_size,bias = False)
    self.hidden_dim  = hidden_size
def forward(self,inputs):
    hx = torch.zeros(1, self.hidden_dim)
    cx = torch.zeros(1, self.hidden_dim)
    hx, cx = self.lstm(inputs,(hx,cx))
    x = hx
    return x,(hx,cx)

hidden_size = 512

lstm = MyModel (hidden_size)

print (lstm.state_dict ())

input = torch.randn (1,10) x, (hx, cx) = lstm (ввод)