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
33
34
35
36
37
38
39
40

/*
    Appellation: linspace <mod>
    Contrib: FL03 <jo3mccain@icloud.com>
*/
use super::step_size;
use num::traits::{FromPrimitive, Num};

pub trait Linspace<T> {
    fn linspace(start: T, stop: T, steps: usize) -> Self;
}

pub trait LinspaceExt<T>: Linspace<T> {
    fn linspace_until(stop: T, steps: usize) -> Self;
}

impl<S, T> LinspaceExt<T> for S
where
    S: Linspace<T>,
    T: Default,
{
    fn linspace_until(stop: T, steps: usize) -> Self {
        S::linspace(T::default(), stop, steps)
    }
}

impl<T> Linspace<T> for Vec<T>
where
    T: Copy + Default + FromPrimitive + Num + PartialOrd,
{
    fn linspace(start: T, stop: T, steps: usize) -> Self {
        let step = step_size(start, stop, steps);
        let mut vec = Vec::with_capacity(steps);
        let mut value = start;
        for _ in 0..steps {
            vec.push(value);
            value = value + step;
        }
        vec
    }
}