dlpark 0.7.0

dlpack Rust binding for Python
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

use super::MemoryLayout;
use crate::ffi;
use crate::ffi::Tensor;

pub trait TensorLike<L>
where
    L: MemoryLayout,
{
    type Error;
    fn data_ptr(&self) -> *mut std::ffi::c_void;
    fn memory_layout(&self) -> L;
    fn device(&self) -> Result<ffi::Device, Self::Error>;
    fn data_type(&self) -> Result<ffi::DataType, Self::Error>;
    fn byte_offset(&self) -> u64;
}

impl Tensor {
    pub fn update<T, L>(&mut self, t: &T, layout: &L) -> Result<(), T::Error>
    where
        T: TensorLike<L>,
        L: MemoryLayout,
    {
        self.data = t.data_ptr();
        self.device = t.device()?;
        self.dtype = t.data_type()?;
        self.byte_offset = t.byte_offset();
        self.ndim = layout.ndim();
        self.shape = layout.shape_ptr();
        self.strides = layout.strides_ptr();
        Ok(())
    }
}