Struct UnifiedSparseTensor 

pub struct UnifiedSparseTensor { /* private fields */ }

Unified sparse tensor that can hold any sparse format

Implementations

impl UnifiedSparseTensor

pub fn new(inner: Box<dyn SparseTensor + Send + Sync>) -> Self

Create a new unified sparse tensor

pub fn from_sparse<T: SparseTensor + Send + Sync + 'static>(tensor: T) -> Self

Create from any sparse tensor type

pub fn inner(&self) -> &dyn SparseTensor

Get the underlying sparse tensor

pub fn metadata_mut(&mut self) -> &mut TensorMetadata

Get mutable access to metadata

pub fn operation_history(&self) -> &[OperationRecord]

Get operation history

pub fn to_dense_cached(&mut self) -> TorshResult<Arc<Tensor>>

Get or compute dense representation with caching

pub fn clear_dense_cache(&mut self)

Clear dense cache to free memory

pub fn optimize_format(&mut self) -> TorshResult<bool>

Convert to optimal format based on access patterns and performance hints

pub fn suggest_optimal_format(&self) -> TorshResult<SparseFormat>

Suggest optimal format based on current metadata

pub fn update_access_pattern( &mut self, operation: &str, region: Option<(usize, usize, usize, usize)>, )

Update access patterns based on operation

pub fn matmul_optimized( &mut self, other: &mut UnifiedSparseTensor, ) -> TorshResult<UnifiedSparseTensor>

Perform matrix multiplication with optimization

pub fn memory_stats(&self) -> MemoryStats

Get memory usage statistics

pub fn optimization_report(&self) -> OptimizationReport

Generate optimization report

Trait Implementations

impl SparseTensor for UnifiedSparseTensor

Implement SparseTensor trait for UnifiedSparseTensor

fn format(&self) -> SparseFormat

Get the sparse format used by this tensor

fn shape(&self) -> &Shape

Get the shape of the tensor

fn dtype(&self) -> DType

Get the data type of the tensor elements

fn device(&self) -> DeviceType

Get the device where the tensor is stored

fn nnz(&self) -> usize

Get the number of non-zero elements

fn to_dense(&self) -> TorshResult<Tensor>

Convert to dense tensor representation

fn to_coo(&self) -> TorshResult<CooTensor>

Convert to COO (Coordinate) format

fn to_csr(&self) -> TorshResult<CsrTensor>

Convert to CSR (Compressed Sparse Row) format

fn to_csc(&self) -> TorshResult<CscTensor>

Convert to CSC (Compressed Sparse Column) format

fn sparsity(&self) -> f32

Calculate sparsity ratio (fraction of zero elements)

fn as_any(&self) -> &dyn Any

Cast as Any for downcasting to concrete types