Struct CandleTensorWrapper 

pub struct CandleTensorWrapper { /* private fields */ }

Wrapper for Candle Tensor to implement TensorLike

Implementations

impl CandleTensorWrapper

pub fn new(tensor: Tensor) -> Self

pub fn inner(&self) -> &Tensor

pub fn into_inner(self) -> Tensor

pub fn from_tensorref(tensor_ref: &TensorRef) -> Option<Tensor>

Safe extraction from Arc

Trait Implementations

impl Clone for CandleTensorWrapper

fn clone(&self) -> CandleTensorWrapper

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for CandleTensorWrapper

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

Formats the value using the given formatter.

impl TensorLike for CandleTensorWrapper

fn to_vec_f32(&self) -> Result<Vec<f32>>

Extract tensor data as Vec - Candle implementation

fn as_any(&self) -> &dyn Any

Downcast support for backend-specific fast paths

fn shape(&self) -> &[usize]

Get tensor shape

fn dtype(&self) -> DataType

Get tensor data type

fn device(&self) -> Device

Get device where tensor resides

fn is_contiguous(&self) -> bool

Check if tensor is contiguous in memory

fn view(&self, start: &[usize], end: &[usize]) -> Result<TensorRef>

Create a view/slice of this tensor

fn reshape(&self, shape: &[usize]) -> Result<TensorRef>

Reshape tensor to new shape (must have same number of elements)

fn to_cpu(&self) -> Result<TensorRef>

Convert tensor to CPU device

fn to_device(&self, device: &Device) -> Result<TensorRef>

Convert tensor to specific device

fn to_dtype(&self, dtype: DataType) -> Result<TensorRef>

Convert tensor to specific data type

fn to_vec_u32(&self) -> Result<Vec<u32>>

Extract tensor data as Vec (for token IDs) This is a convenience method for backends that need to extract token data

fn argmax_last_dim_u32(&self) -> Result<u32>

Fast path: argmax over the last dimension, returning the selected token id.

fn numel(&self) -> usize

Get total number of elements

fn ndim(&self) -> usize

Get number of dimensions

fn is_scalar(&self) -> bool

Check if tensor is scalar (0-dimensional)

fn size_bytes(&self) -> usize

Get size in bytes for this tensor