版权声明:本文为博主原创文章,转载请注明出处。 https://cloud.tencent.com/developer/article/1436696
NTM结构图
lstm作为控制器
图中:A为控制器(为LSTM单元)write为write head, read 为 read_head
源码重点注释:
ops.py中有两个函数,分别为linear() 、 Linear() ,这两个函数有什么区别呢, 看上图, linear
主要用于左下角, Linear用于右上角
build_controller():用于生成LSTM单元,用于NTM的LSTM单元和常见的不同,它由三个部分构成了输入,
build_read_head(): 生成read head, 就是图中的read
参考文献:
1(http://distill.pub/2016/augmented-rnns/#attentional-interfaces)