rstsr-common 0.7.3

An n-Dimension Rust Tensor Toolkit
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
41
42
43
44
45
46
47

use crate::prelude_dev::*;

impl<D> Debug for Layout<D>
where
    D: DimDevAPI,
{
    fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
        let shape = self.shape().as_ref();
        let stride = self.stride().as_ref();
        let offset = self.offset();
        let is_c_contig = self.c_contig();
        let is_f_contig = self.f_contig();
        let is_c_prefer = self.c_prefer();
        let is_f_prefer = self.f_prefer();
        let mut contig = String::new();
        if self.size() == 0 {
            write!(contig, "Empty")?;
        } else {
            if is_c_contig {
                write!(contig, "C")?;
            }
            if is_c_prefer {
                write!(contig, "c")?;
            }
            if is_f_contig {
                write!(contig, "F")?;
            }
            if is_f_prefer {
                write!(contig, "f")?;
            }
            if contig.is_empty() {
                write!(contig, "Custom")?;
            }
        }
        write!(
            f,
            "{}-Dim{}, contiguous: {}\nshape: {:?}, stride: {:?}, offset: {}",
            self.ndim(),
            if self.shape().const_ndim().is_none() { " (dyn)" } else { "" },
            contig,
            shape,
            stride,
            offset,
        )?;
        Ok(())
    }
}