transformer (version 0.2.0)

feed_forward: Feed Forward Layer

Description

Feed Forward Layer

Usage

feed_forward(x, dff, d_model)

Value

output of the feed-forward layer

Arguments

x

inputs

dff

dimensions of feed-forward model

d_model

dimensions of the model