Treatment of `single_token` case.

[erlebach]

Why is single_token treated differently during retrieval but not during storage of memories?
I am trying to understand the logic behind store_memories and retrieve_memories in
the neural_memory.py file.