Function extract_logits 

pub fn extract_logits(
    output_request: &mut OutputRequest,
    input_layer: &Tensor,
    n_vocab: usize,
    n: usize,
)

Extract logits from OutputRequest evaluation