Struct dlpackrs::tensor::ManagedTensor

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

ManagedTensor type with Rust as the main owner of the underlying data.

See DLManagedTensor

Fields

inner: ManagedTensorProxy<'ctx>

Implementations

impl<'tensor, 'ctx> ManagedTensor<'tensor, 'ctx>

pub fn new(tensor: Tensor<'tensor>, manager_ctx: *mut c_void) -> Self

pub fn set_deleter(
    &mut self,
    deleter: unsafe extern "C" fn(_: *mut DLManagedTensor)
)

Sets a deleter function pointer

pub fn into_raw(self) -> *const DLManagedTensor

Consumes the ManagedTensor and returns the raw pointer to its underlying DLManagedTensor.

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

Returns a ManagedTensor instances from a raw pointer to DLManagedTensor.

pub fn into_tensor(self) -> Tensor<'tensor>

Consumes the ManagedTensor and returns Tensor.

Trait Implementations

impl<'tensor, 'ctx: 'tensor> Debug for ManagedTensor<'tensor, 'ctx>

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

Formats the value using the given formatter.