Method __init__

imperative/python/megengine/module/rnn.py:183–184

(self, input_size: int, hidden_size: int, bias: bool = True,)

    """

    def __init__(self, input_size: int, hidden_size: int, bias: bool = True,) -> None:
        super(LSTMCell, self).__init__(input_size, hidden_size, bias, num_chunks=4)

    def forward(
        self, input: Tensor, hx: Optional[Tuple[Tensor, Tensor]] = None