class Model(nn.Module):
    def __init__(self):
        super().__init__()
    
    def forward(self,x):
        m=nn.LSTMCell(x.shape[1],x.shape[1]*2)
        output,_=m(x)
        dense=nn.Linear(output.shape[1],1)