tensr 0.1.3

A high-performance, cross-platform, multi-backend tensor/array library for Rust
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

use crate::{dimension::dim::Dimension, types::DimLen};

pub struct Axes<Dim: Dimension> {
    pub(crate) shape: Dim,
    pub(crate) stride: Dim,
}

impl<Dim: Dimension> Axes<Dim> {
    pub fn new_with_default_stride(shape: Dim) -> Self
    where
        Dim::IndexScalar: std::ops::MulAssign,
    {
        let mut stride = Dim::zero();

        let l = shape.ndim();
        let mut s = Dim::IndexScalar::from(1u16);
        for i in 0..l {
            let j = l - i - 1;

            // Safety: We are constructing a new Dim, so we know that the
            // stride is valid for a contiguous array
            unsafe {
                stride.get_mut()[j as usize] = s.clone();
            }

            s *= shape[j as DimLen].clone();
        }

        Self { shape, stride }
    }
}