in gad/src/array_compare.rs [38:49]
fn max_as(&mut self, v: &af::Array<T>, rdims: af::Dim4) -> Result<af::Array<T>> {
self.check().max_as(&v.dims(), rdims)?;
let vdims = v.dims();
let mut result = v.clone();
for i in 0..4 {
if rdims[i] == vdims[i] {
continue;
}
result = af::max(&result, i as i32);
}
Ok(result)
}