Struct DlpackManagedTensor 

pub struct DlpackManagedTensor { /* private fields */ }

Owned DLPack tensor handle.

Dropping this value will call the DLPack deleter and free the underlying GPU memory.

Implementations

impl DlpackManagedTensor

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

Construct an owned DLPack tensor from a raw pointer.

Safety

ptr must be a valid DLManagedTensor* obtained from a DLPack producer, and ownership must be transferred to the caller (the returned value will call the DLPack deleter on drop).

pub fn as_ptr(&self) -> *mut DLManagedTensor

pub fn into_raw(self) -> *mut DLManagedTensor

Trait Implementations

impl Drop for DlpackManagedTensor

fn drop(&mut self)

Executes the destructor for this type.

impl Send for DlpackManagedTensor

impl Sync for DlpackManagedTensor