call method

@override Tensor call(Tensor tokens)

Implementation

@override
Tensor call(Tensor tokens) {
  var h = embeddings(tokens);
  for (final layer in layers) {
    h = layer(h);
  }
  return h;
}