Struct EdgeFeaturedConfigBuilder 

pub struct EdgeFeaturedConfigBuilder { /* private fields */ }

Implementations

impl EdgeFeaturedConfigBuilder

pub fn node_dim(self, d: usize) -> Self

pub fn edge_dim(self, d: usize) -> Self

pub fn num_heads(self, n: usize) -> Self

pub fn dropout(self, d: f32) -> Self

pub fn concat_heads(self, c: bool) -> Self

pub fn negative_slope(self, s: f32) -> Self

pub fn build(self) -> EdgeFeaturedConfig

Trait Implementations

impl Default for EdgeFeaturedConfigBuilder

fn default() -> EdgeFeaturedConfigBuilder

Returns the “default value” for a type.