Method __init__

imperative/python/megengine/module/rnn.py:538–539

(self, *args, **kwargs)

    """

    def __init__(self, *args, **kwargs) -> None:
        super(LSTM, self).__init__(*args, **kwargs)

    def create_cell(self, layer):
        if layer == 0: