chela 0.0.2

High-performance Machine Learning, Auto-Differentiation and Tensor Algebra crate for Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

use chela::*;
use std::env;

use cpu_time::ProcessTime;


fn profile(size: usize) -> u128 {
    let tensor = NdArray::zeros([size, 2]);
    let mut tensor_slice = tensor.slice(s![.., 0]);

    let start = ProcessTime::now();
    tensor_slice.fill(5_f32);
    start.elapsed().as_nanos()
}

fn main() {
    let args: Vec<String> = env::args().collect();
    let size = if args.len() < 2 { 65536 } else { args[1].parse().unwrap() };

    println!("{}", profile(size));
}