Enum DeepFusionLayer 

pub enum DeepFusionLayer {
    Signal {
        signals: Vec<OperatorTree>,
    },
    Propagate {
        edge_label: Option<String>,
        aggregation: DeepFusionAggregation,
        direction: Direction,
    },
    Conv {
        edge_label: Option<String>,
        hop_weights: Vec<f64>,
        direction: Direction,
    },
    Pool {
        edge_label: Option<String>,
        pool_size: usize,
        method: DeepFusionPoolMethod,
        direction: Direction,
    },
    Flatten,
    Dense {
        weights: Vec<f64>,
        bias: Vec<f64>,
        output_channels: usize,
        input_channels: usize,
    },
    Softmax,
    BatchNorm {
        epsilon: f64,
    },
    Dropout {
        probability: f64,
    },
}

Layer in a OperatorTree::DeepFusion pipeline.

Variants

Signal

Fields

signals: Vec<OperatorTree>

Propagate

Fields

edge_label: Option<String>

aggregation: DeepFusionAggregation

direction: Direction

Conv

Fields

edge_label: Option<String>

hop_weights: Vec<f64>

Self weight followed by one weight per neighbor hop.

direction: Direction

Pool

Fields

edge_label: Option<String>

pool_size: usize

method: DeepFusionPoolMethod

direction: Direction

Flatten

Dense

Fields

weights: Vec<f64>

output_channels x input_channels, row-major.

bias: Vec<f64>

output_channels: usize

input_channels: usize

Softmax

BatchNorm

Fields

epsilon: f64

Dropout

Fields

probability: f64

Trait Implementations

impl Clone for DeepFusionLayer

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.