/tf-attention-example

Simple example of how to do dot-product attention in TensorFlow

Primary LanguagePythonApache License 2.0Apache-2.0

Simple example of how to do scaled dot-product attention in tensorflow.

Install Dependencies

pipenv install

Run tests

pipenv run python test.py