Function tensorflow::ops::arg_max

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

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