Skip to content

Commit d4d8088

Browse files
committed
Implement fold for Axes iterator
1 parent 27c84a5 commit d4d8088

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

src/dimension/axes.rs

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -75,6 +75,15 @@ impl<'a, D> Iterator for Axes<'a, D>
7575
}
7676
}
7777

78+
fn fold<B, F>(self, init: B, f: F) -> B
79+
where
80+
F: FnMut(B, AxisDescription) -> B,
81+
{
82+
(self.start..self.end)
83+
.map(move |i| AxisDescription(Axis(i), self.dim[i], self.strides[i] as isize))
84+
.fold(init, f)
85+
}
86+
7887
fn size_hint(&self) -> (usize, Option<usize>) {
7988
let len = self.end - self.start;
8089
(len, Some(len))

0 commit comments

Comments
 (0)