Struct BlockSparseTensorDescriptor 

pub struct BlockSparseTensorDescriptor<'h> { /* private fields */ }

A block-sparse tensor descriptor (cuTENSOR 2.x). Used on the A operand of a BlockSparseContraction.

Implementations

impl<'h> BlockSparseTensorDescriptor<'h>

pub fn new( handle: &'h Handle, extents: &[i64], block_size: &[i64], strides: Option<&[i64]>, block_indices: &[i32], dtype: DataType, alignment_bytes: u32, ) -> Result<Self>

Build a block-sparse tensor:

• extents — full dense shape
• block_size — size per dim of each non-zero block (same length as extents)
• strides — optional custom strides; None = packed
• block_indices — array of num_modes × block_count ints identifying the non-zero block locations (index per mode per block)

pub fn as_raw(&self) -> cutensorBlockSparseTensorDescriptor_t

Raw cutensorBlockSparseTensorDescriptor_t. Use with care.

Trait Implementations

impl<'h> Debug for BlockSparseTensorDescriptor<'h>

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

Formats the value using the given formatter.

impl Drop for BlockSparseTensorDescriptor<'_>

fn drop(&mut self)

Executes the destructor for this type.

fn pin_drop(self: Pin<&mut Self>)

🔬This is a nightly-only experimental API. (pin_ergonomics)

Execute the destructor for this type, but different to Drop::drop, it requires self to be pinned.