Function conv2d

pub fn conv2d<R: JitRuntime, E: FloatElement>(
    input: JitTensor<R>,
    weight: JitTensor<R>,
    bias: Option<JitTensor<R>>,
    options: ConvOptions<2>,
    strategy: Conv2dStrategy,
) -> Result<JitTensor<R>, ConvLaunchError>

Perform a 2D convolution with the given strategy

• input - The input feature map
• weight - The weights (filter) applied to each kernel
• bias - The bias added to each channel
• options - The options to use for the convolution
• strategy - The convolution algorithm to use. Autotune will pick the fastest available option.