private void validateName(Map<String, TensorType> types, String name, String type) {
        if (!types.containsKey(name)) {
            throw new IllegalArgumentException("Model does not contain required " + type + ": '" + name + "'. " +
                    "Model contains: " + String.join(",", types.keySet()));
        }
    }
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