Struct DetrDecoder 

pub struct DetrDecoder {
    pub layers: Vec<DetrDecoderLayer>,
}

Multi-layer DETR decoder: stacks config.depth decoder layers.

Fields

layers: Vec<DetrDecoderLayer>

Decoder layers in order of application.

Implementations

impl DetrDecoder

pub fn new(cfg: DetrConfig, rng: &mut LcgRng) -> VisionResult<Self>

Build a new DetrDecoder with cfg.depth layers, all Xavier-initialised.

Errors

• DimensionMismatch if cfg.depth == 0.
• Propagates errors from DetrConfig validation (via cloning).

pub fn forward( &self, queries: &[f32], encoder_feats: &[f32], n_enc_tokens: usize, ) -> VisionResult<Vec<f32>>

Apply all decoder layers in sequence.

Parameters

• queries: flat [n_queries × embed_dim].
• encoder_feats: flat [n_enc_tokens × embed_dim].
• n_enc_tokens: number of encoder memory tokens.

Returns

Final queries: flat [n_queries × embed_dim].