def _cell(self, x: torch.Tensor, i: torch.Tensor, states: Tuple[(torch.Tensor, torch.Tensor)]) -> Tuple[(torch.Tensor, torch.Tensor)]:
    'Single time step logic of EA-LSTM cell'
    (h_0, c_0) = states
    gates = self.dynamic_gates(h_0, x)
    (f, o, g) = gates.chunk(3, 1)
    c_1 = ((torch.sigmoid(f) * c_0) + (i * torch.tanh(g)))
    h_1 = (torch.sigmoid(o) * torch.tanh(c_1))
    return (h_1, c_1)
Single time step logic of EA-LSTM cell
def _cell(self, x: torch.Tensor, i: torch.Tensor, states: Tuple[(torch.Tensor, torch.Tensor)]) -> Tuple[(torch.Tensor, torch.Tensor)]:
    
    (h_0, c_0) = states
    gates = self.dynamic_gates(h_0, x)
    (f, o, g) = gates.chunk(3, 1)
    c_1 = ((torch.sigmoid(f) * c_0) + (i * torch.tanh(g)))
    h_1 = (torch.sigmoid(o) * torch.tanh(c_1))
    return (h_1, c_1)