Skip to main content

GraphLayer

torsh_graph

Trait GraphLayer 

Source 
pub trait GraphLayer: Debug {
    // Required methods
    fn forward(&self, graph: &GraphData) -> GraphData;
    fn parameters(&self) -> Vec<Tensor>;
}
Expand description

Trait for graph neural network layers

Required Methods§

Source

fn forward(&self, graph: &GraphData) -> GraphData

Forward pass through the layer

Source

fn parameters(&self) -> Vec<Tensor>

Get layer parameters

Implementors§