in gad/src/array.rs [87:98]
fn sum_as(&mut self, v: &af::Array<T>, rdims: af::Dim4) -> Result<af::Array<T>> {
self.check().sum_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::sum(&result, i as i32);
}
Ok(result)
}