Struct TensorInfo 

pub struct TensorInfo {
    pub data: *mut c_void,
    pub device: DLDevice,
    pub dtype: DLDataType,
    pub shape: Vec<i64>,
    pub strides: Option<Vec<i64>>,
    pub byte_offset: u64,
}

Information about a tensor for DLPack export.

This struct holds all the metadata needed to create a DLPack tensor. Use this with into_dlpack_with_info for explicit control over tensor properties.

Fields

data: *mut c_void

Raw data pointer (device pointer for GPU tensors)

device: DLDevice

Device descriptor

dtype: DLDataType

Data type descriptor

shape: Vec<i64>

Shape (dimensions)

strides: Option<Vec<i64>>

Strides in elements (None for contiguous)

byte_offset: u64

Byte offset from data pointer

Implementations

impl TensorInfo

pub fn contiguous( data: *mut c_void, device: DLDevice, dtype: DLDataType, shape: Vec<i64>, ) -> Self

Create tensor info for a contiguous tensor.

pub fn strided( data: *mut c_void, device: DLDevice, dtype: DLDataType, shape: Vec<i64>, strides: Vec<i64>, ) -> Self

Create tensor info with explicit strides.

Panics

Panics if strides.len() != shape.len(). This invariant is required by DLPack consumers which will read strides[i] for each dimension i.

pub fn with_byte_offset(self, offset: u64) -> Self

Set the byte offset.

Trait Implementations

impl Clone for TensorInfo

fn clone(&self) -> TensorInfo

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for TensorInfo

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

Formats the value using the given formatter.