Struct tract_tensorflow::prelude::Tensor[−]

pub struct Tensor { /* fields omitted */ }
Expand description

Tensor is a concrete tensor in tract.

Implementations

impl Tensor

pub unsafe fn uninitialized<T>(shape: &[usize]) -> Result<Tensor, Error> where
    T: Datum, 

Create an uninitialized tensor (dt as type paramater).

pub unsafe fn uninitialized_dt(
    dt: DatumType,
    shape: &[usize]
) -> Result<Tensor, Error>

Create an uninitialized tensor (dt as regular parameter).

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

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

pub unsafe fn uninitialized_aligned_dt(
    dt: DatumType,
    shape: &[usize],
    alignment: usize
) -> Result<Tensor, Error>

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

pub fn stack_tensors(
    axis: usize,
    tensors: &[impl Borrow<Tensor>]
) -> Result<Tensor, Error>

pub unsafe fn clear<T>(&mut self) where
    T: Datum + Zero, 

pub fn zero<T>(shape: &[usize]) -> Result<Tensor, Error> where
    T: Datum + Zero, 

pub fn zero_scalar<T>() -> Result<Tensor, Error> where
    T: Datum + Zero, 

pub fn zero_scalar_dt(dt: DatumType) -> Result<Tensor, Error>

pub fn zero_dt(dt: DatumType, shape: &[usize]) -> Result<Tensor, Error>

pub fn zero_aligned_dt(
    dt: DatumType,
    shape: &[usize],
    alignment: usize
) -> Result<Tensor, Error>

pub fn zero_aligned<T>(
    shape: &[usize],
    alignment: usize
) -> Result<Tensor, Error> where
    T: Datum + Zero, 

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

Create an tensor from raw data.

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

pub unsafe fn from_raw_aligned<T>(
    shape: &[usize],
    content: &[u8],
    align: usize
) -> Result<Tensor, Error> where
    T: Datum, 

pub unsafe fn from_raw_dt(
    dt: DatumType,
    shape: &[usize],
    content: &[u8]
) -> Result<Tensor, Error>

pub unsafe fn from_raw_dt_align(
    dt: DatumType,
    shape: &[usize],
    content: &[u8],
    align: usize
) -> Result<Tensor, Error>

pub unsafe fn from_slice_align<T>(
    content: &[T],
    align: usize
) -> Result<Tensor, Error> where
    T: Datum, 

pub fn rank(&self) -> usize

Get the number of dimensions (or axes) of the tensor.

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

Notable traits for &'_ [u8]

impl<'_> Read for &'_ [u8]impl<'_> Write for &'_ mut [u8]

Get the shape of the tensor.

pub fn len(&self) -> usize

Get the number of valeus in the tensor.

pub fn strides(&self) -> &[isize]ⓘ

Notable traits for &'_ [u8]

impl<'_> Read for &'_ [u8]impl<'_> Write for &'_ mut [u8]

Get the shape of the tensor.

pub unsafe fn set_shape_unchecked(&mut self, shape: &[usize])

Force the tensor shape, no consistency check.

pub fn set_shape(&mut self, shape: &[usize]) -> Result<(), Error>

Force the tensor shape.

pub fn permute_axes(self, axes: &[usize]) -> Result<Tensor, Error>

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

Reshape the tensor to shape.

pub fn insert_axis(&mut self, axis: usize) -> Result<(), Error>

pub fn remove_axis(&mut self, axis: usize) -> Result<(), Error>

pub fn broadcast_into_rank(self, rank: usize) -> Result<Tensor, Error>

pub fn broadcast_to_rank(&mut self, rank: usize) -> Result<(), Error>

pub fn broadcast_scalar_to_shape(
    &self,
    shape: &[usize]
) -> Result<Tensor, Error>

pub fn assign_slice(
    &mut self,
    range: impl RangeBounds<usize>,
    src: &Tensor,
    src_range: impl RangeBounds<usize>,
    axis: usize
) -> Result<(), Error>

pub unsafe fn assign_slice_unchecked(
    &mut self,
    range: impl RangeBounds<usize>,
    src: &Tensor,
    src_range: impl RangeBounds<usize>,
    axis: usize
)

pub fn datum_type(&self) -> DatumType

Get the datum type of the tensor.

pub unsafe fn set_datum_type(&mut self, dt: DatumType)

Set the datum type of the tensor.

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

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: &Tensor, approx: bool) -> Result<(), Error>

Compare two tensors, allowing for rounding errors.

pub fn into_array<D>(
    self
) -> Result<ArrayBase<OwnedRepr<D>, Dim<IxDynImpl>>, Error> where
    D: Datum, 

Transform the tensor into a ndarray::Array.

pub unsafe fn into_array_unchecked<D>(
    self
) -> ArrayBase<OwnedRepr<D>, Dim<IxDynImpl>> where
    D: Datum, 

Transform the tensor into a ndarray::Array.

pub fn to_array_view<D>(
    &'a self
) -> Result<ArrayBase<ViewRepr<&'a D>, Dim<IxDynImpl>>, Error> where
    D: Datum, 

Transform the data as a ndarray::Array.

pub fn to_array_view_mut<D>(
    &'a mut self
) -> Result<ArrayBase<ViewRepr<&'a mut D>, Dim<IxDynImpl>>, Error> where
    D: Datum, 

Transform the data as a mutable ndarray::Array.

pub unsafe fn to_array_view_unchecked<D>(
    &'a self
) -> ArrayBase<ViewRepr<&'a D>, Dim<IxDynImpl>> where
    D: Datum, 

Transform the data as a ndarray::Array.

pub unsafe fn to_array_view_mut_unchecked<D>(
    &'a mut self
) -> ArrayBase<ViewRepr<&'a mut D>, Dim<IxDynImpl>> where
    D: Datum, 

Transform the data as a mutable ndarray::Array.

pub fn as_ptr<D>(&self) -> Result<*const D, Error> where
    D: Datum, 

Access the data as a pointer.

pub unsafe fn as_ptr_unchecked<D>(&self) -> *const D where
    D: Datum, 

Access the data as a pointer.

pub unsafe fn as_ptr_mut_unchecked<D>(&mut self) -> *mut D where
    D: Datum, 

Access the data as a pointer.

pub fn as_ptr_mut<D>(&mut self) -> Result<*mut D, Error> where
    D: Datum, 

Access the data as a mutable pointer.

pub fn as_slice<D>(&self) -> Result<&[D], Error> where
    D: Datum, 

Access the data as a slice.

pub fn as_slice_mut<D>(&mut self) -> Result<&mut [D], Error> where
    D: Datum, 

Access the data as a mutable slice.

pub unsafe fn as_slice_unchecked<D>(&self) -> &[D]ⓘ

Notable traits for &'_ [u8]

impl<'_> Read for &'_ [u8]impl<'_> Write for &'_ mut [u8]
 where
    D: Datum, 

Access the data as a slice.

pub unsafe fn as_slice_mut_unchecked<D>(&mut self) -> &mut [D]ⓘ

Notable traits for &'_ [u8]

impl<'_> Read for &'_ [u8]impl<'_> Write for &'_ mut [u8]
 where
    D: Datum, 

Access the data as a mutable slice.

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

Access the data as a scalar.

pub unsafe fn to_scalar_unchecked<D>(&'a self) -> &'a D where
    D: Datum, 

Access the data as a scalar.

pub unsafe fn as_bytes(&self) -> &[u8]ⓘ

Notable traits for &'_ [u8]

impl<'_> Read for &'_ [u8]impl<'_> Write for &'_ mut [u8]

pub unsafe fn as_bytes_mut(&mut self) -> &mut [u8]ⓘ

Notable traits for &'_ [u8]

impl<'_> Read for &'_ [u8]impl<'_> Write for &'_ mut [u8]

pub fn is_uniform(&self) -> bool

pub fn as_uniform(&self) -> Option<Tensor>

pub fn cast_to<D>(&self) -> Result<Cow<'_, Tensor>, Error> where
    D: Datum, 

Optionnaly convert data to a tensor for a new DatumType.

pub fn cast_to_dt(&self, dt: DatumType) -> Result<Cow<'_, Tensor>, Error>

Optionnaly convert data to a tensor for a new DatumType.

pub fn cast_to_scalar<D>(&self) -> Result<D, Error> where
    D: Datum + Copy, 

Access the data as a scalar, after a cast.

pub fn nth(&self, nth: usize) -> Result<Tensor, Error>

Access the nth element of the tensor, returned as a 0-rank Tensor

pub fn deep_clone(&self) -> Tensor

pub fn slice(
    &self,
    axis: usize,
    start: usize,
    end: usize
) -> Result<Tensor, Error>

pub fn view(&self) -> TensorView<'_>

pub fn view_at_prefix(&self, prefix: &[usize]) -> Result<TensorView<'_>, Error>

pub fn view_mut(&mut self) -> TensorView<'_>

pub fn view_at_prefix_mut(
    &mut self,
    prefix: &[usize]
) -> Result<TensorView<'_>, Error>

Trait Implementations

impl Clone for Tensor

pub fn clone(&self) -> Tensor

Returns a copy of the value. Read more

fn clone_from(&mut self, source: &Self)1.0.0[src]

Performs copy-assignment from source. Read more

impl Debug for Tensor

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

Formats the value using the given formatter. Read more

impl Default for Tensor

pub fn default() -> Tensor

Returns the “default value” for a type. Read more

impl Drop for Tensor

pub fn drop(&mut self)

Executes the destructor for this type. Read more

impl<'a> From<&'a Tensor> for InferenceFact[src]

pub fn from(t: &'a Tensor) -> InferenceFact[src]

Performs the conversion.

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

pub fn from(t: &'t Tensor) -> TypedFact

Performs the conversion.

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

pub fn from(it: ArrayBase<OwnedRepr<T>, D>) -> Tensor

Performs the conversion.

impl From<Blob> for Tensor

pub fn from(it: Blob) -> Tensor

Performs the conversion.

impl From<String> for Tensor

pub fn from(it: String) -> Tensor

Performs the conversion.

impl From<TDim> for Tensor

pub fn from(it: TDim) -> Tensor

Performs the conversion.

impl From<Tensor> for InferenceFact[src]

pub fn from(t: Tensor) -> InferenceFact[src]

Performs the conversion.

impl From<Tensor> for TypedFact

pub fn from(t: Tensor) -> TypedFact

Performs the conversion.

impl From<bool> for Tensor

pub fn from(it: bool) -> Tensor

Performs the conversion.

impl From<f16> for Tensor

pub fn from(it: f16) -> Tensor

Performs the conversion.

impl From<f32> for Tensor

pub fn from(it: f32) -> Tensor

Performs the conversion.

impl From<f64> for Tensor

pub fn from(it: f64) -> Tensor

Performs the conversion.

impl From<i16> for Tensor

pub fn from(it: i16) -> Tensor

Performs the conversion.

impl From<i32> for Tensor

pub fn from(it: i32) -> Tensor

Performs the conversion.

impl From<i64> for Tensor

pub fn from(it: i64) -> Tensor

Performs the conversion.

impl From<i8> for Tensor

pub fn from(it: i8) -> Tensor

Performs the conversion.

impl From<u16> for Tensor

pub fn from(it: u16) -> Tensor

Performs the conversion.

impl From<u32> for Tensor

pub fn from(it: u32) -> Tensor

Performs the conversion.

impl From<u64> for Tensor

pub fn from(it: u64) -> Tensor

Performs the conversion.

impl From<u8> for Tensor

pub fn from(it: u8) -> Tensor

Performs the conversion.

impl Hash for Tensor

pub fn hash<H>(&self, state: &mut H) where
    H: Hasher, 

Feeds this value into the given Hasher. Read more

fn hash_slice<H>(data: &[Self], state: &mut H) where
    H: Hasher, 1.3.0[src]

Feeds a slice of this type into the given Hasher. Read more

impl IntoArcTensor for Tensor

pub fn into_arc_tensor(self) -> Arc<Tensor>

Convert Self to a Arc. Read more

impl PartialEq<Tensor> for Tensor

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

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

#[must_use]
fn ne(&self, other: &Rhs) -> bool1.0.0[src]

This method tests for !=.

impl<'a> TryFrom<&'a Tensor> for TensorProto[src]

type Error = TractError

The type returned in the event of a conversion error.

fn try_from(from: &Tensor) -> TractResult<TensorProto>[src]

Performs the conversion.

impl<'a> TryFrom<&'a TensorProto> for Tensor[src]

type Error = TractError

The type returned in the event of a conversion error.

fn try_from(t: &TensorProto) -> TractResult<Tensor>[src]

Performs the conversion.

impl Send for Tensor

impl Sync for Tensor

Auto Trait Implementations

Blanket Implementations

impl<T> Any for T where
    T: 'static + ?Sized, [src]

pub fn type_id(&self) -> TypeId[src]

Gets the TypeId of self. Read more

impl<T> Borrow<T> for T where
    T: ?Sized, [src]

pub fn borrow(&self) -> &T[src]

Immutably borrows from an owned value. Read more

impl<T> BorrowMut<T> for T where
    T: ?Sized, [src]

pub fn borrow_mut(&mut self) -> &mut T[src]

Mutably borrows from an owned value. Read more

impl<T> Downcast for T where
    T: Any, 

pub fn into_any(self: Box<T, Global>) -> Box<dyn Any + 'static, Global>ⓘ

Notable traits for Box<R, Global>

impl<R> Read for Box<R, Global> where
    R: Read + ?Sized, impl<W> Write for Box<W, Global> where
    W: Write + ?Sized, impl<F, A> Future for Box<F, A> where
    F: Future + Unpin + ?Sized,
    A: Allocator + 'static,  type Output = <F as Future>::Output;impl<I, A> Iterator for Box<I, A> where
    I: Iterator + ?Sized,
    A: Allocator,  type Item = <I as Iterator>::Item;

Convert Box<dyn Trait> (where Trait: Downcast) to Box<dyn Any>. Box<dyn Any> can then be further downcast into Box<ConcreteType> where ConcreteType implements Trait. Read more

pub fn into_any_rc(self: Rc<T>) -> Rc<dyn Any + 'static>

Convert Rc<Trait> (where Trait: Downcast) to Rc<Any>. Rc<Any> can then be further downcast into Rc<ConcreteType> where ConcreteType implements Trait. Read more

pub fn as_any(&self) -> &(dyn Any + 'static)

Convert &Trait (where Trait: Downcast) to &Any. This is needed since Rust cannot generate &Any’s vtable from &Trait’s. Read more

pub fn as_any_mut(&mut self) -> &mut (dyn Any + 'static)

Convert &mut Trait (where Trait: Downcast) to &Any. This is needed since Rust cannot generate &mut Any’s vtable from &mut Trait’s. Read more

impl<T> DowncastSync for T where
    T: Any + Send + Sync, 

pub fn into_any_arc(self: Arc<T>) -> Arc<dyn Any + 'static + Sync + Send>

Convert Arc<Trait> (where Trait: Downcast) to Arc<Any>. Arc<Any> can then be further downcast into Arc<ConcreteType> where ConcreteType implements Trait. Read more

impl<T> DynClone for T where
    T: Clone, [src]

pub fn __clone_box(&self, Private) -> *mut ()[src]

impl<T> From<T> for T[src]

pub fn from(t: T) -> T[src]

Performs the conversion.

impl<T, U> Into<U> for T where
    U: From<T>, [src]

pub fn into(self) -> U[src]

Performs the conversion.

impl<T> ToOwned for T where
    T: Clone, [src]

type Owned = T

The resulting type after obtaining ownership.

pub fn to_owned(&self) -> T[src]

Creates owned data from borrowed data, usually by cloning. Read more

pub fn clone_into(&self, target: &mut T)[src]

🔬 This is a nightly-only experimental API. (toowned_clone_into)

recently added

Uses borrowed data to replace owned data, usually by cloning. Read more

impl<T, U> TryFrom<U> for T where
    U: Into<T>, [src]

type Error = Infallible

The type returned in the event of a conversion error.

pub fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>[src]

Performs the conversion.

impl<T, U> TryInto<U> for T where
    U: TryFrom<T>, [src]

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.

pub fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>[src]

Performs the conversion.