google-deepmind / dnc

A TensorFlow implementation of the Differentiable Neural Computer.
Apache License 2.0
2.5k stars 443 forks source link

Why the memory is a 3D tensor? #47

Open MrYaoH opened 1 year ago

MrYaoH commented 1 year ago

Dear author: We notice that in the paper, the memory is a 2D matrix (cell number, memory size]), but in pratice, the memory is a 3D tensor ([batch size, cell number, memory size]). Maybe I ignore some import detail. Could you help me address my confusion? Thank you