00748b6ec7a3e845cbfe9319664e357418c0ca73,thinc/layers/lstm.py,,LSTM_step,#Any#Any#,51

Before Change

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


def init(
    model: Model, X: Optional[RNNState] = None, Y: Optional[RNNState] = None

After Change

            "PyTorchWrapper and the torch.LSTM class."
        )
        raise NotImplementedError(msg)
    return Model(
        "lstm_step", forward, init=init, layers=[Linear()], dims={"nO": nO, "nI": nI}
    )


def init(
    model: Model, X: Optional[RNNState] = None, Y: Optional[RNNState] = None

Instances