Enum DeepLayerSpec 

pub enum DeepLayerSpec {
    Input {
        dimensions: usize,
    },
    Embed {
        embedding: Vec<f64>,
    },
    Dense {
        weights: Vec<f64>,
        bias: Vec<f64>,
        output_channels: usize,
        input_channels: usize,
    },
    Flatten,
    GlobalPool {
        method: PoolMethodSpec,
    },
    Softmax,
    BatchNorm {
        epsilon: f64,
    },
    Dropout {
        p: f64,
    },
    CNN1D {
        weights: Vec<f64>,
        bias: Vec<f64>,
        output_channels: usize,
        input_channels: usize,
        kernel_size: usize,
        stride: usize,
        padding: usize,
    },
    CNN2D {
        weights: Vec<f64>,
        bias: Vec<f64>,
        output_channels: usize,
        input_channels: usize,
        input_height: usize,
        input_width: usize,
        kernel_height: usize,
        kernel_width: usize,
        stride_height: usize,
        stride_width: usize,
        padding_height: usize,
        padding_width: usize,
    },
    Propagate {
        edge_label: String,
        aggregation: AggregationSpec,
        direction: DirectionSpec,
    },
    Conv {
        edge_label: String,
        hop_weights: Vec<f64>,
        direction: DirectionSpec,
    },
    Pool {
        edge_label: String,
        pool_size: usize,
        method: PoolKindSpec,
        direction: DirectionSpec,
    },
    Attention,
    RNN {
        weights_input: Vec<f64>,
        weights_hidden: Vec<f64>,
        bias: Vec<f64>,
        hidden_channels: usize,
        input_channels: usize,
        return_sequences: bool,
    },
    LSTM {
        weights_input: Vec<f64>,
        weights_hidden: Vec<f64>,
        bias: Vec<f64>,
        hidden_channels: usize,
        input_channels: usize,
        return_sequences: bool,
    },
}

Variants

Input

Runtime feature-vector input. Used by trained models and batched feature inference.

Fields

dimensions: usize

Embed

Fields

embedding: Vec<f64>

Dense

Fields

weights: Vec<f64>

bias: Vec<f64>

output_channels: usize

input_channels: usize

Flatten

GlobalPool

Fields

method: PoolMethodSpec

Softmax

BatchNorm

Fields

epsilon: f64

Dropout

Fields

p: f64

CNN1D

Fields

weights: Vec<f64>

bias: Vec<f64>

output_channels: usize

input_channels: usize

kernel_size: usize

stride: usize

padding: usize

CNN2D

Fields

weights: Vec<f64>

bias: Vec<f64>

output_channels: usize

input_channels: usize

input_height: usize

input_width: usize

kernel_height: usize

kernel_width: usize

stride_height: usize

stride_width: usize

padding_height: usize

padding_width: usize

Propagate

Fields

edge_label: String

aggregation: AggregationSpec

direction: DirectionSpec

Conv

Fields

edge_label: String

hop_weights: Vec<f64>

direction: DirectionSpec

Pool

Fields

edge_label: String

pool_size: usize

method: PoolKindSpec

direction: DirectionSpec

Attention

RNN

Fields

weights_input: Vec<f64>

weights_hidden: Vec<f64>

bias: Vec<f64>

hidden_channels: usize

input_channels: usize

return_sequences: bool

LSTM

Fields

weights_input: Vec<f64>

weights_hidden: Vec<f64>

bias: Vec<f64>

hidden_channels: usize

input_channels: usize

return_sequences: bool

Trait Implementations

impl Clone for DeepLayerSpec

fn clone(&self) -> DeepLayerSpec

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DeepLayerSpec

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for DeepLayerSpec

fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>

where D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl PartialEq for DeepLayerSpec

fn eq(&self, other: &DeepLayerSpec) -> bool

Equality operator ==.

fn ne(&self, other: &Rhs) -> bool

Inequality operator !=.

impl Serialize for DeepLayerSpec

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

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl StructuralPartialEq for DeepLayerSpec