acme-tensor 0.3.1

A comprehensive tensor library for Rust with support for automatic-differentiation.
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

/*
    Appellation: create <mod>
    Contrib: FL03 <jo3mccain@icloud.com>
*/
pub use self::{arange::*, linspace::*, stack::*, utils::*};

pub(crate) mod arange;
pub(crate) mod linspace;
pub(crate) mod stack;

pub(crate) mod utils {
    use core::ops::{Div, Sub};
    use num::traits::{FromPrimitive, ToPrimitive};
    /// Calculate the step size for a given range and number of steps.
    pub fn step_size<T>(start: T, stop: T, steps: usize) -> T
    where
        T: FromPrimitive + Div<Output = T> + Sub<Output = T>,
    {
        (stop - start) / T::from_usize(steps).unwrap()
    }

    pub fn steps<T>(start: T, stop: T, step: T) -> usize
    where
        T: ToPrimitive + Div<Output = T> + Sub<Output = T>,
    {
        let steps = (stop - start) / step;
        steps.to_usize().unwrap()
    }
}

#[cfg(test)]
mod tests {}