Struct Tensor 

pub struct Tensor { /* private fields */ }

Core tensor abstraction - the single source of truth for all tensor operations

Implementations

impl Tensor

pub fn new( shape: Vec<usize>, dtype: DataType, device: Device, data: Arc<dyn TensorStorage>, ) -> Self

Create new tensor with data

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

Get tensor shape

pub fn dtype(&self) -> DataType

Get tensor data type

pub fn device(&self) -> &Device

Get tensor device

pub fn numel(&self) -> usize

Get number of elements

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

Move tensor to device

pub fn reshape(&self, new_shape: &[usize]) -> Result<Self>

Reshape tensor

pub fn narrow(&self, dim: usize, start: usize, length: usize) -> Result<Self>

Narrow (slice) a tensor along a dimension

Returns a view into the tensor containing elements from start to start + length along the specified dimension.

pub fn from_candle(candle_tensor: CandleTensor) -> Self

Create tensor from Candle tensor

pub fn as_candle(&self) -> Option<&CandleTensor>

Try to get the underlying Candle tensor

pub fn to_candle(&self) -> Result<CandleTensor>

Convert to Candle tensor (may involve copying)

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

Create tensor with zeros using Candle backend

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

Create tensor from f32 slice using Candle backend

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

Create tensor from i64 slice using Candle backend

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

Get the raw data as f32 slice (if applicable)

pub fn transpose(&self) -> Result<Self>

Transpose tensor (swap last two dimensions) For 2D tensors [M, N] -> [N, M] For higher dims [.., M, N] -> [.., N, M]

pub fn transpose_dims(&self, dim0: usize, dim1: usize) -> Result<Self>

Transpose tensor with specific dimensions

Trait Implementations

impl Clone for Tensor

fn clone(&self) -> Tensor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for Tensor

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

Formats the value using the given formatter.