My understanding is that the trick to differentiable key-value stores is to treat the key as a probability over locations, and to treat the value as a weighted

My understanding is that the trick to differentiable key-value stores is to treat the key as a probability over locations, and to treat the value as a weighted sum using these probabilities. From (pdf) https://aclweb.org/anthology/D16-1147 https://aclweb.org/anthology/D16-1147