Struct SqueezeExciteBlock 

pub struct SqueezeExciteBlock {
    pub fc_reduce_w: Tensor,
    pub fc_reduce_b: Tensor,
    pub fc_expand_w: Tensor,
    pub fc_expand_b: Tensor,
    pub channels: usize,
    pub reduced: usize,
}

Squeeze-and-Excitation block (inference-mode).

Channel attention: GlobalAvgPool -> FC(reduce) -> ReLU -> FC(expand) -> Sigmoid -> scale. reduction_ratio controls bottleneck (typically 4 or 16).

Fields

fc_reduce_w: Tensor

fc_reduce_b: Tensor

fc_expand_w: Tensor

fc_expand_b: Tensor

channels: usize

reduced: usize

Implementations

impl SqueezeExciteBlock

pub fn new(channels: usize, reduction_ratio: usize) -> Result<Self, ModelError>

pub fn forward(&self, input: &Tensor) -> Result<Tensor, ModelError>

Forward: input [N,H,W,C] -> scaled [N,H,W,C].