Enum DimensionType

pub enum DimensionType {
    Caffe,
    CaffeC4,
    TensorFlow,
}

The type of the tensor dimension
If you are manually specifying the shapes then this doesn’t really matter
N -> Batch size C -> Channel H -> Height W -> Width

Variants

Caffe

Caffe style dimensions or NCHW

CaffeC4

Caffe style dimensions with channel packed in 4 bytes or NC4HW4

TensorFlow

Tensorflow style dimensions or NHWC

Implementations

impl DimensionType

pub const NHWC: Self = Self::TensorFlow

Tensorflow style dimensions or NHWC

pub const NCHW: Self = Self::Caffe

Caffe style dimensions or NCHW

pub const NC4HW4: Self = Self::CaffeC4

Caffe style dimensions with channel packed in 4 bytes or NC4HW4

Trait Implementations

impl Clone for DimensionType

fn clone(&self) -> DimensionType

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for DimensionType

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

Formats the value using the given formatter.

impl From<DimensionType> for DimensionType

fn from(dm: DimensionType) -> Self

Converts to this type from the input type.

impl Hash for DimensionType

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

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

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for DimensionType

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

Tests for self and other values to be equal, and is used by ==.

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

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Copy for DimensionType

impl Eq for DimensionType

impl StructuralPartialEq for DimensionType