5dccee984b2c25f6508947d3731b8934296ff4c5,thinc/layers/lstm.py,,LSTM_step,#Any#Any#,15

Before Change

        forward,
        init=init,
        dims={"nO": nO, "nI": nI},
        params={"W": None, "b": None, "initial_hiddens": None, "initial_cells": None},
    )

After Change

        "lstm_step", forward, init=init, layers=[Affine()], dims={"nO": nO, "nI": nI}
    )
    if nO is not None and nI is not None:
        model.initialize()
    return model


def init(model, X=None, Y=None):

Instances