Struct DeviceArray 

pub struct DeviceArray<T: GpuDataType, D: Dimension> { /* private fields */ }

Array residing on a specific device (CPU, GPU, TPU)

Implementations

impl<T: GpuDataType, D: Dimension> DeviceArray<T, D>

pub fn device(&self) -> DeviceType

Get the device where the array resides

pub const fn shape(&self) -> &D

Get the shape of the array

pub fn size(&self) -> usize

Get the size of the array in elements

pub fn ndim(&self) -> usize

Get the number of dimensions

pub fn is_on_cpu(&self) -> bool

Check if the array is on the CPU

pub fn is_on_gpu(&self) -> bool

Check if the array is on a GPU

pub fn as_cpuarray(&self) -> Option<&Array<T, IxDyn>>

Get a reference to the underlying CPU array if available

pub fn as_gpubuffer(&self) -> Option<&GpuBuffer<T>>

Get a reference to the underlying GPU buffer if available

impl<T: GpuDataType, D: Dimension> DeviceArray<T, D>

Cross-device array operations

pub fn map<F>( &self, f: F, manager: &DeviceMemoryManager, ) -> CoreResult<DeviceArray<T, D>>

where F: Fn(T) -> T + Send + Sync, D: Clone,

Map the array elements using a function

pub fn reduce<F>(&self, f: F, manager: &DeviceMemoryManager) -> CoreResult<T>

where F: Fn(T, T) -> T + Send + Sync, T: Copy,

Reduce the array using a binary operation

Trait Implementations

impl<T: Clone + GpuDataType, D: Clone + Dimension> Clone for DeviceArray<T, D>

fn clone(&self) -> DeviceArray<T, D>

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<T: Debug + GpuDataType, D: Debug + Dimension> Debug for DeviceArray<T, D>

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

Formats the value using the given formatter.

impl<T, D> ToHost<T, D> for DeviceArray<T, D>

where T: GpuDataType, D: Dimension,

fn to_host(&self, manager: &CrossDeviceManager) -> CoreResult<Array<T, D>>

Transfer the device array to the host