Element-Research / rnn

Recurrent Neural Network library for Torch7's nn
BSD 3-Clause "New" or "Revised" License
941 stars 313 forks source link

Module to add soft/hard attention mechanism to a LSTM #432

Open vguptai opened 6 years ago

vguptai commented 6 years ago

Do we have a module to add attention mechanism to a LSTM. I am doing sequence classification and wanted to explore hard and soft attention mechanism.