Function autograd::ops::reduce_max [] [src]

pub fn reduce_max<T: ArrayLike>(x: &Tensor, axes: &T, keep_dims: bool) -> Tensor

Takes max along specified axes.

Elements of axes can be negative.

extern crate ndarray;
extern crate autograd as ag;

let ref x = ag::constant(ndarray::arr2(&[[2., 4.], [3., 1.]]));
let ref y = ag::reduce_max(&x, &[0], false);

assert_eq!(y.eval(&[]), Ok(ndarray::arr1(&[3., 4.]).into_dyn()));