Struct ManagedTensor

pub struct ManagedTensor(/* private fields */);

ManagedTensor is a wrapper around a dlpack DLManagedTensor object. This lets you pass matrices in device or host memory into cuvs.

Implementations

impl ManagedTensor

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

pub fn to_device(&self, res: &Resources) -> Result<ManagedTensor>

Creates a new ManagedTensor on the current GPU device, and copies the data into it.

pub fn to_host<T: IntoDtype, S: RawData<Elem = T> + RawDataMut, D: Dimension>( &self, res: &Resources, arr: &mut ArrayBase<S, D>, ) -> Result<()>

Copies data from device memory into host memory

Trait Implementations

impl Debug for ManagedTensor

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

Formats the value using the given formatter.

impl Drop for ManagedTensor

fn drop(&mut self)

Executes the destructor for this type.

impl<T: IntoDtype, S: RawData<Elem = T>, D: Dimension> From<&ArrayBase<S, D>> for ManagedTensor

Create a non-owning view of a Tensor from a ndarray

fn from(arr: &ArrayBase<S, D>) -> Self

Converts to this type from the input type.