Struct dlpackrs::tensor::Tensor

#[repr(transparent)]
pub struct Tensor<'tensor> {
    pub inner: DLTensor,
    /* private fields */
}

Non-owned Tensor type interface. See DLTensor

Fields

inner: DLTensor

Implementations

impl<'tensor> Tensor<'tensor>

pub fn new(
    data: *mut c_void,
    device: Device,
    ndim: i32,
    dtype: DataType,
    shape: *mut i64,
    strides: *mut i64,
    byte_offset: u64
) -> Self

Constructor

pub fn into_inner(self) -> DLTensor

Returns the underlying DLTensor where lifetime parameter is removed.

pub fn into_raw(self) -> *const DLTensor

Consumes the Tensor and returns the raw pointer to its underlying DLTensor.

pub unsafe fn from_raw(ptr: *mut DLTensor) -> Self

Creates a Tensor from a raw DLTensor pointer (must be non-null).

pub fn data(&self) -> *mut c_void

Returns a *mut pointer to the underlying data of the Tensor.

pub fn device(&self) -> Device

Returns the device type.

pub fn itemsize(&self) -> usize

Returns the size of an entry/item in the Tensor.

pub fn ndim(&self) -> usize

Returns the number of dimensions of the Tensor.

pub fn dtype(&self) -> DataType

Returns the type of the entries of the Tensor.

pub fn shape(&self) -> Option<&[usize]>

Returns the shape of the Tensor.

pub fn strides(&self) -> Option<&[usize]>

Returns the strides of the underlying Tensor.

pub fn byte_offset(&self) -> isize

Returns the byte offset of the underlying Tensor.

pub fn size(&self) -> Option<usize>

Returns the size of the memory required to store the underlying data of the Tensor.

Trait Implementations

impl<'tensor> Debug for Tensor<'tensor>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<'tensor> From<DLTensor> for Tensor<'tensor>

fn from(dts: DLTensor) -> Self

Converts to this type from the input type.

impl<'tensor> From<Tensor<'tensor>> for DLTensor

fn from(ts: Tensor<'tensor>) -> Self

Converts to this type from the input type.