Enum TensorTransformation 

pub enum TensorTransformation {
    Transpose {
        dim0: usize,
        dim1: usize,
    },
    Reshape {
        shape: Vec<usize>,
    },
    Cast {
        dtype: DataType,
    },
    Quantize {
        config: QuantizationConfig,
    },
    Scale {
        factor: f32,
    },
    Slice {
        dim: usize,
        start: Option<usize>,
        end: Option<usize>,
    },
}

Transformations that can be applied to loaded tensors

Variants

Transpose

Transpose dimensions

Fields

dim0: usize

dim1: usize

Reshape

Reshape to new shape

Fields

shape: Vec<usize>

Cast

Convert data type

Fields

dtype: DataType

Quantize

Quantize tensor

Fields

config: QuantizationConfig

Scale

Apply scaling

Fields

factor: f32

Slice

Slice tensor

Fields

dim: usize

start: Option<usize>

end: Option<usize>

Trait Implementations

impl Clone for TensorTransformation

fn clone(&self) -> TensorTransformation

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TensorTransformation

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for TensorTransformation

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

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for TensorTransformation

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

where __S: Serializer,

Serialize this value into the given Serde serializer.