CyberZHG / keras-self-attention

Attention mechanism for processing sequential data that considers the context for each timestamp.

Home Page:https://pypi.org/project/keras-self-attention/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

在加法模式和乘法模式里,一个加 ba,一个加ba[0]

xugaoliang opened this issue · comments

ba的shape为(1,),在加法模式中是直接加 self.ba,在乘法模式里却是加 self.ba[0],感觉应该没必要取[0]

这个跟兼容theano有关,去掉会报错。