Add the option to use attention as the state. Is already learning something,...

Add the option to use attention as the state. Is already learning something, but in a very slow way....