memory.py: fix indexing for read_modes transform

ixaxaar / pytorch-dnc

Differentiable Neural Computers, Sparse Access Memory and Sparse Differentiable Neural Computers, for Pytorch

MIT License

335 stars 56 forks source link

Closed jbinas closed 6 years ago

jbinas commented 6 years ago

The read_modes vector overlaps with write_gate, and should be shifted by one element.

ixaxaar commented 6 years ago

LGTM

Thank you so much for the fix!