Struct TensorDescriptor 

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

A tensor descriptor: modes + extents + dtype + stride layout.

Implementations

impl<'h> TensorDescriptor<'h>

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

extents[i] is the size along mode i. strides = None gets a row-major packed layout.

pub fn as_raw(&self) -> cutensorTensorDescriptor_t

Raw cutensorTensorDescriptor_t. Use with care.

pub unsafe fn set_attribute( &self, attr: i32, buf: *const c_void, size_bytes: usize, ) -> Result<()>

Low-level tensor-descriptor attribute setter.

Safety

buf must point at size_bytes matching attr.

Trait Implementations

impl<'h> Debug for TensorDescriptor<'h>

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.