LSTM¶
Computes an one-layer LSTM. This operator is usually supported via some custom implementation such as CuDNN. Notations: * X - input tensor * i - input gate * o - output gate * f - forget gate * c - cell gate * t - time step (t-1 means previous time step) * W[iofc] - W parameter weight …
Abstract Signature:
LSTM(X: Tensor, W: Tensor, R: Tensor, B: Tensor, sequence_lens, initial_h: Tensor, initial_c: Tensor, P: Tensor, activation_alpha: List[float], activation_beta: List[float], activations: List[str], clip: float, direction: str, hidden_size: int, input_forget: int, layout: int)