Trait dfdx::devices::ReduceElements
source · [−]pub trait ReduceElements<T: CountElements> {
fn reduce<F: FnMut(T::Dtype, T::Dtype) -> T::Dtype + Copy>(
inp: &T,
f: F
) -> T::Dtype;
fn sum(inp: &T) -> T::Dtype
where
T::Dtype: Num,
{ ... }
fn mean(inp: &T) -> T::Dtype
where
T::Dtype: Float + FromPrimitive,
{ ... }
fn max(inp: &T) -> T::Dtype
where
T::Dtype: Float,
{ ... }
fn min(inp: &T) -> T::Dtype
where
T::Dtype: Float,
{ ... }
}
Expand description
Reduce an entire Nd array to 1 value