Why the memory is a 3D tensor?

[MrYaoH]

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