Expand description
Unified tensor operations and device management Core Tensor Abstraction
This module provides the foundational tensor abstraction that unifies all tensor operations across CPU, GPU, and different data formats.
Uses Candle (HuggingFace’s Rust ML framework) as the computational backend.
Modules§
- ops_fn
- Convenience functions for tensor operations
Structs§
- Candle
Storage - Candle-backed tensor storage - the primary storage implementation
- CpuStorage
- CPU tensor storage implementation
- CpuTensor
OpsImpl - CPU implementation of tensor operations
- GpuStorage
- GPU tensor storage implementation (legacy - kept for compatibility)
- GpuTensor
OpsImpl - Basic GPU implementation of tensor operations
- Tensor
- Core tensor abstraction - the single source of truth for all tensor operations
- Tensor
Dispatcher - Tensor operation dispatcher - automatically selects CPU/GPU implementation
Enums§
Traits§
- Tensor
Ops - Core tensor operations trait - device-agnostic interface
- Tensor
Storage - Tensor storage trait - abstracts how data is actually stored
Functions§
- ops
- Get global tensor operations