represent method

Tensor represent(
  1. List<int> observations
)

Implementation

Tensor represent(List<int> observations) {
  // We pass a dummy encoder output (zeros) because AFT expects it
  final dummyEnc = Tensor.zeros([1, embedSize]);
  return transformer.getLatentState(observations, dummyEnc);
}