Struct ManagedTensor

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

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

See DLManagedTensor

Fields

inner: ManagedTensorProxy<C>

Implementations

impl<'tensor, C: 'tensor> ManagedTensor<'tensor, C>

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

Contructor.

pub fn set_deleter(&mut self, deleter: fn(&mut ManagedTensor<'_, C>))

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, C: Debug + 'tensor> Debug for ManagedTensor<'tensor, C>

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

Formats the value using the given formatter.

impl<'tensor, C> From<DLManagedTensor> for ManagedTensor<'tensor, C>

fn from(dlm: DLManagedTensor) -> Self

Converts to this type from the input type.

impl<'tensor, C> From<ManagedTensor<'tensor, C>> for DLManagedTensor

fn from(mt: ManagedTensor<'tensor, C>) -> Self

Converts to this type from the input type.