Skip to content

First working version of the Attention Layer

Pre-release
Pre-release
Compare
Choose a tag to compare
@thushv89 thushv89 released this 11 Apr 10:19
· 45 commits to master since this release

This commit contains the current working version of the Attention layer. This has the following functionality.

  • Works with Keras layers such as LSTM, GRU and Bidirectional layers. However the this version only supports if a fixed batch size is specified in the model (i.e. use batch_shape to specify shape in layers instead of shape)