Function arg_max

pub fn arg_max<O0: Into<Output>, O1: Into<Output>>(
    input: O0,
    dimension: O1,
    scope: &mut Scope,
) -> Result<Operation>

Shorthand for ArgMax::new().build(input, dimension, scope).