forward method

Tensor forward(
  1. Tensor x,
  2. List<Tensor> tracker
)

Implementation

Tensor forward(Tensor x, List<Tensor> tracker) {
  // 1. Expand and Activate (GELU)
  final hidden = w1.forward(x, tracker);
  // 2. Contract back to original dimension
  final out = w2.forward(hidden, tracker);
  return out;
}