Struct TensorDescriptor 

pub struct TensorDescriptor<'comm> { /* private fields */ }

Implementations

impl<'comm> TensorDescriptor<'comm>

pub fn create_for<T: DataTypeLike>( context: &Context<'comm>, shape: &[u64], ranks_per_mode: &[u64], rank_count: u32, ) -> Result<Self>

pub fn create( context: &Context<'comm>, config: TensorDescriptorConfig<'_>, ) -> Result<Self>

pub unsafe fn from_raw( handle: cutensorMpTensorDescriptor_t, context: &Context<'comm>, config: TensorDescriptorConfig<'_>, ) -> Result<Self>

Wraps an existing cuTENSORMp tensor descriptor.

Safety

handle must be a valid cuTENSORMp tensor descriptor associated with context. The metadata in config must describe the raw descriptor. The returned value takes ownership of handle and destroys it on drop.

pub fn rank(&self) -> u32

pub fn shape(&self) -> &[u64]

pub fn element_stride(&self) -> Option<&[u64]>

pub fn block_size(&self) -> Option<&[u64]>

pub fn block_stride(&self) -> Option<&[u64]>

pub fn ranks_per_mode(&self) -> &[u64]

pub fn ranks(&self) -> Option<&[i32]>

pub fn rank_count(&self) -> u32

pub fn data_type(&self) -> DataType

pub const fn as_raw(&self) -> cutensorMpTensorDescriptor_t

pub fn into_raw(self) -> cutensorMpTensorDescriptor_t

Consumes this descriptor and returns the owned raw cuTENSORMp tensor descriptor.

The caller becomes responsible for destroying the descriptor.

Trait Implementations

impl<'comm> Debug for TensorDescriptor<'comm>

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

Formats the value using the given formatter.

impl Drop for TensorDescriptor<'_>

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.