Struct Tensor 

pub struct Tensor { /* private fields */ }

ABI stable owned Shape for ffi

Implementations

impl Tensor

pub fn data_ptr(&self) -> *const c_void

Get the data pointer of the Tensor

Returns

• *mut core::ffi::c_void - The data pointer of the Tensor

pub fn data_ptr_mut(&mut self) -> *mut c_void

Get the data pointer of the Tensor

Returns

• *mut core::ffi::c_void - The data pointer of the Tensor

pub fn is_contiguous(&self) -> bool

Check if the Tensor is contiguous

Returns

• bool - True if the Tensor is contiguous, false otherwise

pub fn data_as_slice<T: AsDLDataType>(&self) -> Result<&[T]>

pub fn data_as_slice_mut<T: AsDLDataType>(&self) -> Result<&mut [T]>

Returns the tensor data as a mutable slice.

This method takes &self rather than &mut self by design: like std::fs::File::write, the metadata of a Tensor (shape, dtype, device) is governed by Rust’s ownership rules, but writing to the underlying data buffer (CPU memory or a GPU pointer) is a side-effect outside Rust’s aliasing model. Most C/CUDA kernel APIs accept a non-mut Tensor and mutate its data content, so requiring &mut self here would force artificial mutability annotations throughout the deep-learning stack with no real safety benefit.

Safety contract (caller responsibility)

If the Tensor has been cloned (via ObjectArc), the caller must ensure no other clone is concurrently reading the data.

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

pub fn ndim(&self) -> usize

pub fn numel(&self) -> usize

pub fn strides(&self) -> &[i64]

pub fn dtype(&self) -> DLDataType

pub fn device(&self) -> DLDevice

impl Tensor

pub fn from_nd_alloc<TNDAlloc>( alloc: TNDAlloc, shape: &[i64], dtype: DLDataType, device: DLDevice, ) -> Self

where TNDAlloc: NDAllocator,

Arguments

• alloc - The NDAllocator
• shape - The shape of the Tensor
• dtype - The data type of the Tensor
• device - The device of the Tensor

Returns

• Tensor - The created Tensor

pub fn from_slice<T: AsDLDataType>(slice: &[T], shape: &[i64]) -> Result<Self>

Create a Tensor from a slice

Arguments

• slice - The slice to create the Tensor from
• shape - The shape of the Tensor

Returns

• Tensor - The created Tensor

Trait Implementations

impl AnyCompatible for Tensor

fn type_str() -> String

the type string of the type

unsafe fn copy_to_any_view(src: &Self, data: &mut TVMFFIAny)

the value to copy to TVMFFIAny

unsafe fn check_any_strict(data: &TVMFFIAny) -> bool

unsafe fn copy_from_any_view_after_check(data: &TVMFFIAny) -> Self

Copy value from TVMFFIAny after checking caller must ensure that the value is compatible with the type

unsafe fn move_to_any(src: Self, data: &mut TVMFFIAny)

consume the value to move to Any

unsafe fn move_from_any_after_check(data: &mut TVMFFIAny) -> Self

the value to move from TVMFFIAny NOTE: pay very careful attention to avoid memory leak!

unsafe fn try_cast_from_any_view(data: &TVMFFIAny) -> Result<Self, ()>

try to cast the value from AnyView

fn get_mismatch_type_info(data: &TVMFFIAny) -> String

Get the type key of a type when TryCastFromAnyView fails.

impl<'a> ArgIntoRef for &'a Tensor

type Target = Tensor

fn to_ref(&self) -> &Self::Target

impl ArgIntoRef for Tensor

type Target = Tensor

fn to_ref(&self) -> &Self::Target

impl Clone for Tensor

fn clone(&self) -> Tensor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<'a> IntoArgHolder for &'a Tensor

type Target = &'a Tensor

fn into_arg_holder(self) -> Self::Target

impl IntoArgHolder for Tensor

type Target = Tensor

fn into_arg_holder(self) -> Self::Target

impl ObjectRefCore for Tensor

type ContainerType = <ObjectArc<TensorObj> as Deref>::Target

fn data(this: &Self) -> &ObjectArc<Self::ContainerType>

fn into_data(this: Self) -> ObjectArc<Self::ContainerType>

fn from_data(data: ObjectArc<Self::ContainerType>) -> Self

impl TryFrom<Any> for Tensor

type Error = Error

The type returned in the event of a conversion error.

fn try_from(value: Any) -> Result<Self, Self::Error>

Performs the conversion.

impl<'a> TryFrom<AnyView<'a>> for Tensor

type Error = Error

The type returned in the event of a conversion error.

fn try_from(value: AnyView<'a>) -> Result<Self, Self::Error>

Performs the conversion.