Skip to main content

conv_forward

burn_cubecl::kernel::conv

Function conv_forward 

Source 
pub fn conv_forward<const N: usize>(
    input: CubeTensor,
    weight: CubeTensor,
    bias: Option<CubeTensor>,
    options: ConvOptions<N>,
    strategy: ConvStrategy,
) -> Result<CubeTensor, ConvSetupError>
Expand description

Performs an N-dimensional 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.