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 mut ctx = ag::Context::new(); let ref x = ag::constant(ndarray::arr2(&[[2., 4.], [3., 1.]]), &mut ctx); let ref y = ag::reduce_max(&x, &[0], false); assert_eq!(y.eval(&mut ctx), ndarray::arr1(&[3., 4.]).into_dyn());