candela-tensor 0.2.0

A lazy, graph-based tensor engine in Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

use crate::{branch_fast_iter, tensor::storage::TensorData};

#[inline]
pub(crate) fn clone_to_buffer<T: Clone>(tensor: &TensorData<T>, mut buffer: Vec<T>) -> Vec<T> {
    branch_fast_iter!(tensor.fast_iter() => iter, {
        for (i, el) in iter.cloned().enumerate() {
            buffer[i] = el;
        }

        buffer
    })
}

#[inline]
pub(crate) fn normalize_axis(axis: isize, shape_len: usize) -> usize {
    (if axis < 0 {
        shape_len as isize + axis
    } else {
        axis
    }) as usize
}