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 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

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.