Struct DecLayer 

pub struct DecLayer { /* private fields */ }

Implementations

impl DecLayer

pub fn load( vb: VarBuilder<'_>, config: &ProteinMPNNConfig, layer: i32, ) -> Result<Self>

The main decoder stack: num_in = 3 * hidden_dim.

pub fn load_with_num_in( vb: VarBuilder<'_>, config: &ProteinMPNNConfig, layer: i32, num_in: usize, ) -> Result<Self>

num_in explicitly, for LigandMPNN’s context_encoder_layers.

Those are DecLayer(hidden_dim, hidden_dim * 2) in the reference — the message is [h_V_C, h_E_context, Y_nodes] rather than the decoder’s three hidden-width blocks — so their W1 is (128, 384) where the decoder’s is (128, 512) (ferritin-100.11).

pub fn forward( &self, h_v: &Tensor, h_e: &Tensor, mask_v: Option<&Tensor>, mask_attend: Option<&Tensor>, training: Option<bool>, ) -> Result<Tensor>

Trait Implementations

impl Clone for DecLayer

fn clone(&self) -> DecLayer

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DecLayer

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

Formats the value using the given formatter.