Struct GraphAttentionConfigBuilder 

pub struct GraphAttentionConfigBuilder { /* private fields */ }

Builder for GraphAttentionConfig.

Implementations

impl GraphAttentionConfigBuilder

pub fn dim(self, dim: usize) -> Self

Sets the model dimension.

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

Sets the number of attention heads.

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

Sets the edge feature dimension.

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

Sets the negative slope for LeakyReLU.

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

Sets whether to concatenate multi-head outputs.

pub fn build(self) -> AttentionResult<GraphAttentionConfig>

Builds the GraphAttentionConfig.

Trait Implementations

impl Default for GraphAttentionConfigBuilder

fn default() -> GraphAttentionConfigBuilder

Returns the “default value” for a type.