private static Tensor createTensorRepresentation(List<Integer> tokens, String dimension) {
        var size = tokens.size();
        TensorType type = new TensorType.Builder(TensorType.Value.FLOAT).indexed(dimension, size).build();
        IndexedTensor.Builder builder = IndexedTensor.Builder.of(type);
        for (int i = 0; i < size; ++i) {
            builder.cell(tokens.get(i), i);
        }
        return builder.build();
    }
Given a tensor 'logits' with 3 dimensions: batch, sequence, and vocabulary
find the value in the vocabulary dimension with highest score for the given
token in the sequence
model-integration/src/main/java/ai/vespa/llm/clients/OnnxEncoderDecoder.java