Skip to main content

select_output

Function select_output 

Source
pub fn select_output<'a>(
    outputs: &[&'a TensorDyn],
    config: &SemanticConfig,
) -> DecoderResult<&'a TensorDyn>
Expand description

Picks the tensor of a semantic output from a model’s outputs: the one named like the output, else the only one with its shape, else the one at the output’s position when every model output was passed.