private static Tensor createAttentionMask(Tensor d) {
        return d.map((x) -> x > 0 ? 1 : 0);
    }
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