Enum TensorValue 

pub enum TensorValue {
    F32(Array<f32, IxDyn>),
    I64(Array<i64, IxDyn>),
}

Variants

F32(Array<f32, IxDyn>)

I64(Array<i64, IxDyn>)

Implementations

impl TensorValue

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

pub fn into_f32(self) -> Option<Array<f32, IxDyn>>

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

Trait Implementations

impl Clone for TensorValue

fn clone(&self) -> TensorValue

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TensorValue

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

Formats the value using the given formatter.

impl From<ArrayBase<OwnedRepr<f32>, Dim<IxDynImpl>>> for TensorValue

fn from(arr: Array<f32, IxDyn>) -> Self

Converts to this type from the input type.

impl From<ArrayBase<OwnedRepr<i64>, Dim<IxDynImpl>>> for TensorValue

fn from(arr: Array<i64, IxDyn>) -> Self

Converts to this type from the input type.