Tensor represent(List<int> observations) { return transformer.getLatentState(observations, Tensor.zeros([1, embedSize])); }