Struct Tensor

#[repr(C)]
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]>

Get the data as a mutable slice

Note that we do allow mutable data access to copies of the Tensor, as in the case of low-level deep learning frameworks.

Arguments

• T - The type of the data

Returns

• Result<&mut [T]> - The data as a mutable slice

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.