Enum OperationType 

pub enum OperationType {
    Linear {
        input_dim: usize,
        output_dim: usize,
    },
    Convolution {
        filters: usize,
        kernel_size: usize,
    },
    GraphConv {
        channels: usize,
        aggregation: String,
    },
    Attention {
        heads: usize,
        embed_dim: usize,
    },
    Transformer {
        layers: usize,
        heads: usize,
    },
    Embedding {
        vocab_size: usize,
        embed_dim: usize,
    },
    Activation {
        function: String,
    },
    Normalization {
        method: String,
    },
    Dropout {
        rate: f32,
    },
    SkipConnection,
    Pooling {
        method: String,
        size: usize,
    },
    Custom {
        operation_id: String,
        params: HashMap<String, f32>,
    },
}

Types of operations available for architecture building

Variants

Linear

Fields

input_dim: usize

output_dim: usize

Convolution

Fields

filters: usize

kernel_size: usize

GraphConv

Fields

channels: usize

aggregation: String

Attention

Fields

heads: usize

embed_dim: usize

Transformer

Fields

layers: usize

heads: usize

Embedding

Fields

vocab_size: usize

embed_dim: usize

Activation

Fields

function: String

Normalization

Fields

method: String

Dropout

Fields

rate: f32

SkipConnection

Pooling

Fields

method: String

size: usize

Custom

Fields

operation_id: String

params: HashMap<String, f32>

Trait Implementations

impl Clone for OperationType

fn clone(&self) -> OperationType

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for OperationType

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for OperationType

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

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for OperationType

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

where __S: Serializer,

Serialize this value into the given Serde serializer.