Struct CandleStorage 

pub struct CandleStorage { /* private fields */ }

Candle-backed tensor storage - the primary storage implementation

Implementations

impl CandleStorage

pub fn new(tensor: CandleTensor) -> Self

Create from a Candle tensor

pub fn tensor(&self) -> &CandleTensor

Get the underlying Candle tensor

pub fn tensor_mut(&mut self) -> &mut CandleTensor

Get a mutable reference to the Candle tensor

pub fn zeros(shape: &[usize], dtype: DataType, device: &Device) -> Result<Self>

Create zeros tensor with Candle

pub fn ones(shape: &[usize], dtype: DataType, device: &Device) -> Result<Self>

Create ones tensor with Candle

pub fn from_f32_slice( data: &[f32], shape: &[usize], device: &Device, ) -> Result<Self>

Create tensor from raw f32 data

pub fn from_i64_slice( data: &[i64], shape: &[usize], device: &Device, ) -> Result<Self>

Create tensor from raw i64 data

Trait Implementations

impl Clone for CandleStorage

fn clone(&self) -> CandleStorage

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for CandleStorage

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

Formats the value using the given formatter.

impl TensorStorage for CandleStorage

fn data_ptr(&self) -> *const u8

Get raw data pointer (for unsafe operations)

fn data_ptr_mut(&mut self) -> *mut u8

Get mutable data pointer (for unsafe operations)

fn size_bytes(&self) -> usize

Get data size in bytes

fn device(&self) -> &Device

Get device where data is stored

fn to_device(&self, device: &Device) -> Result<Arc<dyn TensorStorage>>

Copy data to another device

fn clone_storage(&self) -> Arc<dyn TensorStorage>

Clone the storage

fn as_candle_tensor(&self) -> Option<&CandleTensor>

Try to get the underlying Candle tensor (if this is CandleStorage)