Skip to content

Plug-and-Play version of Attention implemented in Tensorflow

Notifications You must be signed in to change notification settings

dhruvramani/tf.attention

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 

Repository files navigation

Attention in Tensorflow

Plug-and-Play version of Attention implemented in Tensorflow (1.2.1).

Installation

Install Tensorflow for your system.

Usage

You can use this file with your code in the following manner :

    from attention import Attention
    ...
    output_vectors = lstm()
    atta = Attention(output_vectors)
    att_vec = atta.applyAttention()
    output = tf.nn.softmax(tf.matmul(Wh, att_vec) + b)
    ...

Current Status

Haven't checked, will be running it on something soon, hopefully.

About

Plug-and-Play version of Attention implemented in Tensorflow

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages