Struct CudaSparseTensor 

pub struct CudaSparseTensor {
    pub format: SparseFormat,
    pub device_id: i32,
    pub dtype: DType,
    pub shape: [usize; 2],
    pub nnz: usize,
    /* private fields */
}

CUDA sparse tensor wrapper

Fields

format: SparseFormat

The sparse format used on GPU

device_id: i32

Device index

dtype: DType

Data type

shape: [usize; 2]

Shape of the tensor

nnz: usize

Number of non-zero elements

Implementations

impl CudaSparseTensor

pub fn from_coo(coo: &CooTensor, device_id: i32) -> TorshResult<Self>

Create a new CUDA sparse tensor from COO format

pub fn from_csr(csr: &CsrTensor, device_id: i32) -> TorshResult<Self>

Create a new CUDA sparse tensor from CSR format

pub fn to_coo_gpu(&self) -> TorshResult<CudaSparseTensor>

Convert to COO format on GPU

pub fn to_csr_gpu(&self) -> TorshResult<CudaSparseTensor>

Convert to CSR format on GPU

pub fn spmm(&self, dense: &Tensor) -> TorshResult<Tensor>

Sparse matrix multiplication on GPU

pub fn spgemm(&self, other: &CudaSparseTensor) -> TorshResult<CudaSparseTensor>

Sparse matrix-sparse matrix multiplication on GPU

pub fn to_cpu(&self) -> TorshResult<CooTensor>

Copy tensor to CPU

pub fn is_cuda_available() -> bool

Check if CUDA is available

pub fn memory_usage(&self) -> usize

Get device memory usage

pub fn nnz(&self) -> usize

Get number of non-zero elements

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

Get tensor shape

pub fn dtype(&self) -> DType

Get data type

pub fn device_id(&self) -> i32

Get device ID

pub fn format(&self) -> SparseFormat

Get sparse format

Trait Implementations

impl Clone for CudaSparseTensor

fn clone(&self) -> CudaSparseTensor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for CudaSparseTensor

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

Formats the value using the given formatter.