elasticai.creator.base_modules.lstm_cell

Module Contents

Classes

MathOperations
LSTMCell

API

class elasticai.creator.base_modules.lstm_cell.MathOperations

Bases: elasticai.creator.base_modules.math_operations.Quantize, elasticai.creator.base_modules.math_operations.Add, elasticai.creator.base_modules.math_operations.MatMul, elasticai.creator.base_modules.math_operations.Mul, typing.Protocol

class elasticai.creator.base_modules.lstm_cell.LSTMCell(input_size: int, hidden_size: int, bias: bool, operations: elasticai.creator.base_modules.lstm_cell.MathOperations, sigmoid_factory: collections.abc.Callable[[], torch.nn.Module], tanh_factory: collections.abc.Callable[[], torch.nn.Module], device: Any = None)

Bases: torch.nn.Module

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