Struct CandleTensorFactory 

pub struct CandleTensorFactory { /* private fields */ }

Candle tensor factory

Implementations

impl CandleTensorFactory

pub fn new(device: Device) -> Self

Trait Implementations

impl Debug for CandleTensorFactory

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

Formats the value using the given formatter.

impl TensorFactory for CandleTensorFactory

fn empty( &self, shape: &[usize], dtype: DataType, device: Device, ) -> Result<TensorRef>

创建指定形状/数据类型的空张量（[MVP]）

fn from_slice( &self, data: &[f32], shape: &[usize], dtype: DataType, device: Device, ) -> Result<TensorRef>

通过 slice 数据创建张量（[MVP]）

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

迁移张量到目标设备（[MVP]）

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

执行窄视图操作（[MVP]）

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

reshape 张量（[MVP]）

fn zeros_like(&self, tensor: &TensorRef) -> Result<TensorRef>

基于已有张量创建零填充张量（[MVP]）

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

Create tensor filled with zeros（[Phase 2+] 可选实现）

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

Create tensor filled with ones（[Phase 2+]）

fn uniform( &self, shape: &[usize], low: f32, high: f32, dtype: DataType, device: &Device, ) -> Result<TensorRef>

Create tensor from uniform random distribution（[Phase 2+]）

fn normal( &self, shape: &[usize], mean: f32, std: f32, dtype: DataType, device: &Device, ) -> Result<TensorRef>

Create tensor from normal distribution（[Phase 2+]）

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

Create tensor from existing tensor reference (may involve copying)