Skip to content

Commit 56774bf

Browse files
closure to remove clones
1 parent 2f0a09d commit 56774bf

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

src/numeric/impl_numeric.rs

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -137,13 +137,14 @@ where
137137
}
138138

139139
let mut res = Array::ones(self.raw_dim());
140-
let mut running_product = Array::ones(self.raw_dim().remove_axis(axis));
140+
let running_product = Array::ones(self.raw_dim().remove_axis(axis));
141141

142142
Zip::from(self.axis_iter(axis))
143143
.and(res.axis_iter_mut(axis))
144-
.for_each(|view, mut res| {
145-
running_product = &running_product * &view;
144+
.fold(running_product, |mut running_product, view, mut res| {
145+
running_product = running_product * &view;
146146
res.assign(&running_product);
147+
running_product
147148
});
148149

149150
res

0 commit comments

Comments
 (0)