LSTMCell¶
- class
dragon.vm.torch.nn.
LSTMCell
(
input_size,
hidden_size,
bias=True
)[source]¶ Apply a long short-term memory (LSTM) cell. [Hochreiter & Schmidhuber, 1997].
Examples:
m = torch.nn.LSTMCell(32, 64) x = torch.ones(32, 256) h1, c1 = m(x)