Feed forward layer implemented in Keras
Project description
Implementation of position-wise feed forward layer in the paper: Attention is All You Need
Install
pip install keras-position-wise-feed-forward
Usage
import keras
from keras_position_wise_feed_forward import FeedForward
input_layer = keras.layers.Input()
feed_forward_layer = FeedForward()(input_layer)
model = keras.models.Model(inputs=input_layer, outputs=feed_forward_layer)
model.compile(optimizer='adam', loss='mse', metrics={})
model.summary()
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Close
Hashes for keras-position-wise-feed-forward-0.1.0.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | de754649bea73b5d87cf870649c737caca08165a1eb9652475e589a833810d7c |
|
MD5 | 00c386e61b8a241a4375779f65a7b912 |
|
BLAKE2b-256 | 0a9afc9e4ffea64752d43576f180cf69b5fbfeedb3d6c5386b6565d708b58b18 |