Enum tfdeploy::tensor::Tensor

pub enum Tensor {
    F32(ArrayD<f32>),
    F64(ArrayD<f64>),
    I32(ArrayD<i32>),
    I8(ArrayD<i8>),
    U8(ArrayD<u8>),
    TDim(ArrayD<TDim>),
    String(ArrayD<i8>),
}

Variants

F32(ArrayD<f32>)F64(ArrayD<f64>)I32(ArrayD<i32>)I8(ArrayD<i8>)U8(ArrayD<u8>)TDim(ArrayD<TDim>)String(ArrayD<i8>)

Methods

impl Tensor

pub fn from_pb(t: &TensorProto) -> Result<Tensor>

pub fn from_content<T: Copy, V: Copy>(     dims: Vec<usize>,     content: &[V] ) -> Result<ArrayD<T>>

pub fn to_pb(&self) -> Result<TensorProto>

Important traits for &'a [u8]

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

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

pub fn reduce(&mut self)

pub fn datum_type(&self) -> DatumType

pub fn axis_chunks(&self, axis: usize, size: usize) -> Result<Vec<Tensor>>

pub fn axis_chunks_t<T: Datum>(     &self,     axis: usize,     size: usize ) -> Result<Vec<Tensor>>

pub fn partial_dump(&self, _single_line: bool) -> Result<String>

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

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

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

pub fn cast_to_array<D: Datum>(&self) -> Result<MaybeOwnedArray<D>>

impl Tensor

pub fn as_f64(&self) -> Option<f64>

pub fn as_f64s(&self) -> Option<&ArrayD<f64>>

pub fn take_f64s(self) -> Option<ArrayD<f64>>

pub fn f64s(shape: &[usize], values: &[f64]) -> Result<Tensor>

impl Tensor

pub fn as_f32(&self) -> Option<f32>

pub fn as_f32s(&self) -> Option<&ArrayD<f32>>

pub fn take_f32s(self) -> Option<ArrayD<f32>>

pub fn f32s(shape: &[usize], values: &[f32]) -> Result<Tensor>

impl Tensor

pub fn as_i32(&self) -> Option<i32>

pub fn as_i32s(&self) -> Option<&ArrayD<i32>>

pub fn take_i32s(self) -> Option<ArrayD<i32>>

pub fn i32s(shape: &[usize], values: &[i32]) -> Result<Tensor>

impl Tensor

pub fn as_u8(&self) -> Option<u8>

pub fn as_u8s(&self) -> Option<&ArrayD<u8>>

pub fn take_u8s(self) -> Option<ArrayD<u8>>

pub fn u8s(shape: &[usize], values: &[u8]) -> Result<Tensor>

impl Tensor

pub fn as_i8(&self) -> Option<i8>

pub fn as_i8s(&self) -> Option<&ArrayD<i8>>

pub fn take_i8s(self) -> Option<ArrayD<i8>>

pub fn i8s(shape: &[usize], values: &[i8]) -> Result<Tensor>

impl Tensor

pub fn as_dim(&self) -> Option<TDim>

pub fn as_dims(&self) -> Option<&ArrayD<TDim>>

pub fn take_dims(self) -> Option<ArrayD<TDim>>

pub fn dims(shape: &[usize], values: &[TDim]) -> Result<Tensor>

Trait Implementations

impl IntoExp<ValueFact> for Tensor

fn bex(self) -> Exp<ValueFact>

Converts the value to an Expression.

impl Output for Tensor

fn into_wrapped(source: Tensor) -> Wrapped

Wraps the fact in the Wrapped type.

fn from_wrapped(wrapped: Wrapped) -> Result<Tensor>

Retrieves the fact from the Wrapped type. Panics if wrapped doesn't have the right constructor.

fn wrap(self) -> Wrapped

Wraps self in the Wrapped 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 PartialEq for Tensor

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

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

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

This method tests for !=.

impl Debug for Tensor

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

Formats the value using the given formatter.

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

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

Performs the conversion.

impl From<f64> for Tensor

fn from(it: f64) -> Tensor

Performs the conversion.

impl From<f32> for Tensor

fn from(it: f32) -> Tensor

Performs the conversion.

impl From<i32> for Tensor

fn from(it: i32) -> Tensor

Performs the conversion.

impl From<u8> for Tensor

fn from(it: u8) -> Tensor

Performs the conversion.

impl From<i8> for Tensor

fn from(it: i8) -> Tensor

Performs the conversion.

impl From<TDim> for Tensor

fn from(it: TDim) -> Tensor

Performs the conversion.