Method conv2d

cake-core/src/backends/mod.rs:658–676

(
        &self,
        x: &Tensor,
        weight: &Tensor,
        bias: Option<&Tensor>,
        padding: usize,
        stride: usize,
        dilation: usize,
        groups: usize,
    )

    /// Input: `(batch, in_channels, height, width)`, weight: `(out_channels, in_channels/groups, kH, kW)`.
    #[allow(clippy::too_many_arguments)]
    fn conv2d(
        &self,
        x: &Tensor,
        weight: &Tensor,
        bias: Option<&Tensor>,
        padding: usize,
        stride: usize,
        dilation: usize,
        groups: usize,
    ) -> Result<Tensor> {
        let out = x.conv2d(weight, padding, stride, dilation, groups)?;
        match bias {
            Some(b) => {
                let b = b.reshape((1, b.dim(0)?, 1, 1))?;
                out.broadcast_add(&b)
            }
            None => Ok(out),
        }
    }

    // ── Device control ───────────────────────────────────────────────