FeedForward constructor

FeedForward(
  1. int embedDim,
  2. int ffnDim
)

Implementation

FeedForward(int embedDim, int ffnDim) {
  fc1 = Linear(embedDim, ffnDim);
  fc2 = Linear(ffnDim, embedDim);
}