Struct tract_core::ops::nn::BaseDataShape

pub struct BaseDataShape<D, S> where
    D: DimLike,
    S: AsRef<[D]> + Debug,  {
    pub fmt: DataFormat,
    pub shape: S,
    pub strides: TVec<D>,
}

Fields

fmt: DataFormat

shape: S

strides: TVec<D>

Methods

impl<D, S> BaseDataShape<D, S> where
    D: DimLike,
    S: AsRef<[D]> + Debug, 

pub fn rank(&self) -> usize

pub fn hw_rank(&self) -> usize

pub fn n_axis(&self) -> usize

pub fn c_axis(&self) -> usize

pub fn h_axis(&self) -> usize

pub fn hw_axes(&self) -> Range<usize>

pub fn n_dim(&self) -> &D

pub fn c_dim(&self) -> &D

pub fn hw_dims(&self) -> &[D]

pub fn n(&self) -> &D

pub fn c(&self) -> &D

pub fn n_stride(&self) -> &D

pub fn hw_strides(&self) -> &[D]

pub fn w_stride(&self) -> &D

pub fn c_stride(&self) -> &D

Trait Implementations

impl<D: Clone, S: Clone> Clone for BaseDataShape<D, S> where
    D: DimLike,
    S: AsRef<[D]> + Debug, 

fn clone(&self) -> BaseDataShape<D, S>

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<D: PartialEq, S: PartialEq> PartialEq<BaseDataShape<D, S>> for BaseDataShape<D, S> where
    D: DimLike,
    S: AsRef<[D]> + Debug, 

fn eq(&self, other: &BaseDataShape<D, S>) -> bool

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

fn ne(&self, other: &BaseDataShape<D, S>) -> bool

This method tests for !=.

impl<D: Debug, S: Debug> Debug for BaseDataShape<D, S> where
    D: DimLike,
    S: AsRef<[D]> + Debug, 

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

Formats the value using the given formatter.