Enum DataLayout 

pub enum DataLayout {
    NCHW,
    NHWC,
    NCDHW,
    NDHWC,
    NCL,
    NLC,
    Auto,
}

Supported data layouts for tensors

Variants

NCHW

Channels first: [N, C, H, W] - Optimal for GPU/CUDA

NHWC

Channels last: [N, H, W, C] - Optimal for CPU/NEON

NCDHW

Channels first for 3D: [N, C, D, H, W]

NDHWC

Channels last for 3D: [N, D, H, W, C]

NCL

Channels first for 1D: [N, C, L]

NLC

Channels last for 1D: [N, L, C]

Auto

Auto-detect optimal layout based on device and operation

Implementations

impl DataLayout

pub fn ndim(&self) -> usize

Get the number of dimensions for this layout

pub fn channel_axis(&self) -> usize

Get channel axis for this layout

pub fn is_channels_first(&self) -> bool

Check if this is a channels-first layout

pub fn to_permutation(&self, target: DataLayout) -> Option<Vec<usize>>

Get the permutation indices to convert from this layout to target layout

Trait Implementations

impl Clone for DataLayout

fn clone(&self) -> DataLayout

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DataLayout

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

Formats the value using the given formatter.

impl Hash for DataLayout

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 DataLayout

fn eq(&self, other: &DataLayout) -> 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 DataLayout

impl Eq for DataLayout

impl StructuralPartialEq for DataLayout