Skip to main content

conv_plain

Function conv_plain 

Source
pub fn conv_plain<D: Float>(
    c_in: usize,
    c_out: usize,
    k: usize,
    s: usize,
) -> impl Fn(SymTensor) -> SymTensor
Expand description

Plain Conv2d with bias — no BatchNorm, no activation.

Matches nn.Conv2d(c_in, c_out, k, bias=True) from the ultralytics Detect head.