`elasticai.creator.base_modules.lstm`#

Module Contents#

Classes#

`LayerFactory`
`LSTM`

API#

class elasticai.creator.base_modules.lstm.LayerFactory[source]#

Bases: typing.Protocol

lstm(input_size: int, hidden_size: int, bias: bool) → elasticai.creator.base_modules.lstm_cell.LSTMCell[source]#

class elasticai.creator.base_modules.lstm.LSTM(input_size: int, hidden_size, bias: bool, batch_first: bool, layers: elasticai.creator.base_modules.lstm.LayerFactory)[source]#

Bases: torch.nn.Module

property hidden_size: int#

property input_size: int#

forward(x: torch.Tensor, state: Optional[tuple[torch.Tensor, torch.Tensor]] = None) → tuple[torch.Tensor, tuple[torch.Tensor, torch.Tensor]][source]#