Lstm Function In Python