mdarray 0.7.2

Multidimensional array for Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

//! Expression module, for multidimensional iteration.

mod adapters;
mod buffer;
mod expression;
mod into_expr;
mod iter;
mod sources;

pub use adapters::{Cloned, Copied, Enumerate, Map, Zip, cloned, copied, enumerate, map, zip};
pub use buffer::{Buffer, Drain};
pub use expression::{Apply, Expression, FromExpression, IntoExpression};
pub use into_expr::IntoExpr;
pub use iter::Iter;
pub use sources::{AxisExpr, AxisExprMut, Lanes, LanesMut};
pub use sources::{Fill, FillWith, FromElem, FromFn, fill, fill_with, from_elem, from_fn};

/// Folds all elements of the argument into an accumulator by applying an operation,
/// and returns the result.
///
/// # Examples
///
/// ```
/// use mdarray::{expr, view};
///
/// let v = view![0, 1, 2];
///
/// assert_eq!(expr::fold(v, 0, |acc, x| acc + x), 3);
/// ```
#[inline]
pub fn fold<T, I: IntoExpression, F: FnMut(T, I::Item) -> T>(expr: I, init: T, f: F) -> T {
    expr.into_expr().fold(init, f)
}

/// Calls a closure on each element of the argument.
///
/// # Examples
///
/// ```
/// use mdarray::{expr, tensor, view};
///
/// let mut t = tensor![0, 1, 2];
///
/// expr::for_each(&mut t, |x| *x *= 2);
///
/// assert_eq!(t, view![0, 2, 4]);
/// ```
#[inline]
pub fn for_each<I: IntoExpression, F: FnMut(I::Item)>(expr: I, f: F) {
    expr.into_expr().for_each(f);
}