forward method

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

Implementation

Tensor forward(List<int> tokens, List<Tensor> tracker) {
  // Uses your token embedding + position embedding + transformer blocks
  return encoder.forward(tokens, tracker);
}