Struct TensorDescriptor 

pub struct TensorDescriptor { /* private fields */ }

Implementations

impl TensorDescriptor

pub fn create_for<T: DataTypeLike>( context: &Context, shape: &[u64], devices: &[TensorDevice], ) -> Result<Self>

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

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

Wraps an existing cuTENSORMg tensor descriptor.

Safety

handle must be a valid cuTENSORMg 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 device_count(&self) -> Option<&[i32]>

pub fn devices(&self) -> &[TensorDevice]

pub fn data_type(&self) -> DataType

pub const fn as_raw(&self) -> cutensorMgTensorDescriptor_t

pub fn into_raw(self) -> cutensorMgTensorDescriptor_t

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

The caller becomes responsible for destroying the descriptor.

Trait Implementations

impl Debug for TensorDescriptor

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.