
class LSTM:

    def __init__(self):
        pass

if __name__ == "__main__":
    print("Training controller...")

    # Initial input is an empty embeddings

    # After that, the input to the controller is the output of the previous step