Enum WeightTransform 

pub enum WeightTransform {
    Identity,
    Transpose(Vec<usize>),
    Reshape(Vec<isize>),
    Split {
        axis: usize,
        sizes: Vec<usize>,
    },
    Merge {
        axis: usize,
    },
    ConvFormat {
        from: ConvFormat,
        to: ConvFormat,
    },
}

Transformations that may be needed when converting weights

Variants

Identity

No transformation

Transpose(Vec<usize>)

Transpose specific dimensions

Reshape(Vec<isize>)

Reshape to new dimensions

Split

Split into multiple tensors

Fields

axis: usize

sizes: Vec<usize>

Merge

Merge multiple tensors

Fields

axis: usize

ConvFormat

Convert convolution weights format

Fields

from: ConvFormat

to: ConvFormat

Trait Implementations

impl Clone for WeightTransform

fn clone(&self) -> WeightTransform

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for WeightTransform

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for WeightTransform

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for WeightTransform

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.