FeedForward constructor

FeedForward(
  1. int dModel, {
  2. int hiddenMult = 4,
})

Implementation

FeedForward(int dModel, {int hiddenMult = 4})
    : fc1 = Linear(dModel, dModel * hiddenMult),
      fc2 = Linear(dModel * hiddenMult, dModel);