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
22

use crate::dtype::NumericDataType;
use crate::{NdArray, RawDataType, StridedMemory};
use num::NumCast;

impl<T: NumericDataType> NdArray<'_, T> {
    pub fn astype<'b, F: NumericDataType>(&self) -> NdArray<'b, F>
    {
        let mut data = vec![F::default(); self.size()];

        for (dst, src) in data.iter_mut().zip(self.flatiter()) {
            *dst = NumCast::from(src).expect("astype conversion failed");
        }
        
        unsafe { NdArray::from_contiguous_owned_buffer(self.shape().to_vec(), data) }
    }
}

impl<'a, T: RawDataType> AsRef<NdArray<'a, T>> for NdArray<'a, T> {
    fn as_ref(&self) -> &NdArray<'a, T> {
        self
    }
}