Struct tract_core::tensor::Tensor

pub struct Tensor { /* fields omitted */ }

Tensor is a concrete tensor in tract.

Methods

impl Tensor

pub unsafe fn uninitialized_aligned<T: Datum>(
    shape: &[usize],
    alignment: usize
) -> TractResult<Tensor>

Create an uninitialized tensor with a given alignment (in bytes).

pub unsafe fn from_raw<T: Datum>(
    shape: &[usize],
    content: &[u8]
) -> TractResult<Tensor>

Create an tensor from raw data.

It copies the data, aligning it to the size of T.

pub fn into_aligned(self, alignment: usize) -> TractResult<Tensor>

Re-align a tensor to a byte alignment.

pub unsafe fn null<T: Datum>(shape: &[usize]) -> TractResult<Tensor>

Creates a null tensor (this is rare, and should stay that way).

pub unsafe fn null_dt(dt: DatumType, shape: &[usize]) -> TractResult<Tensor>

Creates a null tensor (this is rare, and should stay that way).

pub fn is_null(&self) -> bool

Check weather self is a null tensor.

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

Get the shape of the tensor.

pub unsafe fn into_shape(self, shape: &[usize]) -> TractResult<Tensor>

Reshape the tensor to shape.

pub fn datum_type(&self) -> DatumType

Get the datum type of the tensor.

pub fn dump_t<D: Datum>(&self, force_full: bool) -> TractResult<String>

Dump the tensor in a human readable form.

force_full will force the tensor to be dump in full even if it is big.

pub fn dump(&self, force_full: bool) -> TractResult<String>

Dump the tensor in a human readable form.

force_full will force the tensor to be dump in full even if it is big.

pub fn close_enough(&self, other: &Self, approx: bool) -> bool

Compare two tensors, allowing for rounding errors.

pub fn into_array<D: Datum>(self) -> TractResult<ArrayD<D>>

Transform the tensor into a ndarray::Array.

pub fn to_array_view<'a, D: Datum>(&'a self) -> TractResult<ArrayViewD<'a, D>>

Transform the data as a ndarray::Array.

pub fn to_array_view_mut<'a, D: Datum>(
    &'a mut self
) -> TractResult<ArrayViewMutD<'a, D>>

Transform the data as a mutable ndarray::Array.

pub fn as_ptr<D: Datum>(&self) -> TractResult<*const D>

Access the data as a pointer.

pub fn as_ptr_mut<D: Datum>(&mut self) -> TractResult<*mut D>

Access the data as a mutable pointer.

pub fn as_slice<D: Datum>(&self) -> TractResult<&[D]>

Access the data as a slice.

pub fn as_slice_mut<D: Datum>(&mut self) -> TractResult<&mut [D]>

Access the data as a mutable slice.

pub fn to_scalar<'a, D: Datum>(&'a self) -> TractResult<&D>

Access the data as a scalar.

pub fn cast_to<D: Datum>(&self) -> TractResult<Cow<Tensor>>

Optionnaly convert data to a tensor for a new DatumType.

pub fn cast_to_dt(&self, dt: DatumType) -> TractResult<Cow<Tensor>>

Optionnaly convert data to a tensor for a new DatumType.

Trait Implementations

impl IntoArcTensor for Tensor

fn into_arc_tensor(self) -> Arc<Tensor>

Convert Self to a Arc.

impl PartialEq<Tensor> for Tensor

fn eq(&self, other: &Tensor) -> bool

This method tests for self and other values to be equal, and is used by ==.

#[must_use]

fn ne(&self, other: &Rhs) -> bool

This method tests for !=.

impl Default for Tensor

fn default() -> Tensor

Returns the "default value" for a type.

impl Clone for Tensor

fn clone(&self) -> Tensor

Returns a copy of the value.

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

Performs copy-assignment from source.

impl From<bool> for Tensor

fn from(it: bool) -> Tensor

Performs the conversion.

impl From<f16> for Tensor

fn from(it: f16) -> Tensor

Performs the conversion.

impl From<f32> for Tensor

fn from(it: f32) -> Tensor

Performs the conversion.

impl From<f64> for Tensor

fn from(it: f64) -> Tensor

Performs the conversion.

impl From<i8> for Tensor

fn from(it: i8) -> Tensor

Performs the conversion.

impl From<i16> for Tensor

fn from(it: i16) -> Tensor

Performs the conversion.

impl From<i32> for Tensor

fn from(it: i32) -> Tensor

Performs the conversion.

impl From<i64> for Tensor

fn from(it: i64) -> Tensor

Performs the conversion.

impl From<u8> for Tensor

fn from(it: u8) -> Tensor

Performs the conversion.

impl From<u16> for Tensor

fn from(it: u16) -> Tensor

Performs the conversion.

impl From<TDim> for Tensor

fn from(it: TDim) -> Tensor

Performs the conversion.

impl From<String> for Tensor

fn from(it: String) -> Tensor

Performs the conversion.

impl From<Tensor> for TypedTensorInfo

fn from(t: Tensor) -> TypedTensorInfo

Performs the conversion.

impl<'t> From<&'t Tensor> for TypedTensorInfo

fn from(t: &'t Tensor) -> TypedTensorInfo

Performs the conversion.

impl<D: Dimension, T: Datum> From<ArrayBase<OwnedRepr<T>, D>> for Tensor

fn from(it: Array<T, D>) -> Tensor

Performs the conversion.

impl Debug for Tensor

fn fmt(&self, formatter: &mut Formatter) -> Result

Formats the value using the given formatter.